diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-16 21:40:18 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-16 21:40:18 +0800 |
| commit | e77d74684a5b77b4b5ce1edac67518fea12bd5c1 (patch) | |
| tree | 2f20ae9ee71ac2584d0e2e484e70f9113a191392 | |
| parent | 6a24467e5c1417a9e41db583ea11cc72d561669c (diff) | |
| parent | f9f45c8d32c6f99c314f1ca93140598d1005c8fb (diff) | |
Merge branch 'python3.8' into python3.7
| -rw-r--r-- | models/model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/models/model.py b/models/model.py index 8cb1459..85f81b4 100644 --- a/models/model.py +++ b/models/model.py @@ -51,8 +51,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 |