summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-03-16 21:40:10 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-03-16 21:40:10 +0800
commitf9f45c8d32c6f99c314f1ca93140598d1005c8fb (patch)
tree2005b02cd6c5cf6e7182c2e713a197fd036ac29a
parent03d38fe32d668c28d7cba01f0b2e227f32b954c1 (diff)
parent38555617816cfaef6f330e7fc90f3cfa65d692fb (diff)
Merge branch 'master' into python3.8
-rw-r--r--models/model.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/models/model.py b/models/model.py
index fa56057..573b1e6 100644
--- a/models/model.py
+++ b/models/model.py
@@ -54,8 +54,8 @@ class Model:
self.hp = hyperparameter_config
self.curr_iter = self.meta.get('restore_iter', 0)
self.total_iter = self.meta.get('total_iter', 80_000)
- self.curr_iters = self.meta.get('restore_iters', (0, 0, 0))
- self.total_iters = self.meta.get('total_iters', (80000, 80000, 80000))
+ self.curr_iters = self.meta.get('restore_iters', (self.curr_iter,))
+ self.total_iters = self.meta.get('total_iters', (self.total_iter,))
self.is_train: bool = True
self.in_channels: int = 3