Update UModel.py

2021-10-21 10:40:01 +09:00 · 2021-10-21 10:40:01 +09:00 · 1211aaae26
commit 1211aaae26
parent 8842af01f7
1 changed files with 2 additions and 0 deletions
--- a/UModel.py
+++ b/UModel.py
@ -95,6 +95,8 @@ class UModel(nn.Module):

    def _load(self, load_best=False):
        path = None
+        self.trained_epochs = 0
+        self.best_cer = 1.0
        if load_best == True and os.path.exists(self.checkpoint + '_best.pth'):
            path = path = self.checkpoint + '_best.pth'
        elif os.path.exists(self.checkpoint + '_last.pth'):