diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 20:43:54 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 20:43:54 +0800 |
| commit | 478ecf50f3dd9e2c2f8ca62e6ecd4a65b2cc7c3a (patch) | |
| tree | 891a10c5fa03fb752fb420ca2f4a647c5b83eae3 /models/model.py | |
| parent | 30b475c0a27e0f848743abf0f909607defc6a3ee (diff) | |
| parent | 3a69cb5325b4a8a51b75224906d839863adeddd6 (diff) | |
Merge branch 'python3.8' into data_parallel_py3.8
# Conflicts:
# models/model.py
Diffstat (limited to 'models/model.py')
| -rw-r--r-- | models/model.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/models/model.py b/models/model.py index 91d6651..07ef37e 100644 --- a/models/model.py +++ b/models/model.py @@ -291,7 +291,7 @@ class Model: 'PartNet': losses[4] }, self.curr_iter) # None-zero losses in batch - if hpm_num_non_zero is not None and hpm_num_non_zero is not None: + if hpm_num_non_zero is not None and pn_num_non_zero is not None: self.writer.add_scalars('Loss/non-zero counts', { 'HPM': hpm_num_non_zero.mean(), 'PartNet': pn_num_non_zero.mean() @@ -470,7 +470,6 @@ class Model: @staticmethod def evaluate( - self, gallery_samples: Dict[str, Dict[str, Union[List, torch.Tensor]]], probe_samples: Dict[str, Dict[str, Union[List, torch.Tensor]]], num_ranks: int = 5 |