diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-13 10:59:59 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-13 10:59:59 +0800 |
| commit | 8ee391b65e2b48d777a268749f54b3aa9e4b9142 (patch) | |
| tree | 85bea16f0c95722ae3c3b37c5ebd0d50800432e6 /utils/configuration.py | |
| parent | abb6989683829f12bf43ff19580444bf3396ac44 (diff) | |
Add multiple checkpoints for different model and set default config value
Diffstat (limited to 'utils/configuration.py')
| -rw-r--r-- | utils/configuration.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/utils/configuration.py b/utils/configuration.py index f1b5d5a..9a8c2ae 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -64,6 +64,8 @@ class ModelConfiguration(TypedDict): name: str restore_iter: int total_iter: int + restore_iters: tuple[int, ...] + total_iters: tuple[int, ...] class Configuration(TypedDict): |