From 8ee391b65e2b48d777a268749f54b3aa9e4b9142 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Wed, 13 Jan 2021 10:59:59 +0800
Subject: Add multiple checkpoints for different model and set default config
 value

---
 utils/configuration.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'utils/configuration.py')

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):
-- 
cgit v1.2.3