summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-03-12 20:43:54 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-03-12 20:43:54 +0800
commit478ecf50f3dd9e2c2f8ca62e6ecd4a65b2cc7c3a (patch)
tree891a10c5fa03fb752fb420ca2f4a647c5b83eae3
parent30b475c0a27e0f848743abf0f909607defc6a3ee (diff)
parent3a69cb5325b4a8a51b75224906d839863adeddd6 (diff)
Merge branch 'python3.8' into data_parallel_py3.8
# Conflicts: # models/model.py
-rw-r--r--models/model.py3
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