summaryrefslogtreecommitdiff
path: root/models/model.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-19 20:36:22 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-19 20:36:22 +0800
commitd4d2154ff886da65f7c2ca429a24e9c3e9666dee (patch)
treee89e24147b8b625050a27e67a52b5d0066435dd4 /models/model.py
parente9de937bf0d72f00f2cb0c56f60b87aa838a0613 (diff)
parent4aa9044122878a8e2b887a8b170c036983431559 (diff)
Merge branch 'master' into data_parallel
Diffstat (limited to 'models/model.py')
-rw-r--r--models/model.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/models/model.py b/models/model.py
index c7ed4b2..4f24ac5 100644
--- a/models/model.py
+++ b/models/model.py
@@ -404,7 +404,7 @@ class Model:
for (iter_, (condition, selector)) in zip(
iters, dataset_selectors.items()
):
- self.curr_iter = self.total_iter = iter_
+ self.curr_iter = iter_
self._dataset_sig = self._make_signature(
dict(**dataset_config, **selector),
popped_keys=['root_dir', 'cache_on']