diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 20:37:00 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 20:37:00 +0800 |
| commit | 3a69cb5325b4a8a51b75224906d839863adeddd6 (patch) | |
| tree | 71e4309bd28f5808bc4333ceab9d1e0e74b14668 /models/model.py | |
| parent | c5b4e984a6830fa2d7abe1183eef02ec00a15433 (diff) | |
| parent | 5c4483ea0c7b2e9166f94d6b6e3b8a070eb959f2 (diff) | |
Merge branch 'master' into python3.8
Diffstat (limited to 'models/model.py')
| -rw-r--r-- | models/model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/model.py b/models/model.py index 9b73f32..4015524 100644 --- a/models/model.py +++ b/models/model.py @@ -273,7 +273,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() |