From 38555617816cfaef6f330e7fc90f3cfa65d692fb Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 16 Mar 2021 15:42:06 +0800 Subject: Set *_iter as *_iters in default --- models/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/model.py b/models/model.py index c1cc703..36a4f7f 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 -- cgit v1.2.3