summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-03-12 20:37:17 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-03-12 20:37:17 +0800
commitc97bed6721fc1a7186719fef0c340d6d03057891 (patch)
tree97e2de760038e913073076f0444b0c0720b1b89a
parent2457ac03ca7dee9962d9a8bcb3e0ad0f44ce185a (diff)
parent3a69cb5325b4a8a51b75224906d839863adeddd6 (diff)
Merge branch 'python3.8' into python3.7
-rw-r--r--models/model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/models/model.py b/models/model.py
index acc78d3..b11d92a 100644
--- a/models/model.py
+++ b/models/model.py
@@ -270,7 +270,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()