fix typos

zhou13 · Jul 9, 2019 · 437ef29 · 437ef29
1 parent 7aee92c
commit 437ef29
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lcnn/trainer.py b/lcnn/trainer.py
@@ -144,16 +144,16 @@ def validate(self):
                 "model_state_dict": self.model.state_dict(),
                 "best_mean_loss": self.best_mean_loss,
             },
-            osp.join(self.out, "checkpoint_lastest.pth.tar"),
+            osp.join(self.out, "checkpoint_latest.pth.tar"),
         )
         shutil.copy(
-            osp.join(self.out, "checkpoint_lastest.pth.tar"),
+            osp.join(self.out, "checkpoint_latest.pth.tar"),
             osp.join(npz, "checkpoint.pth.tar"),
         )
         if self.mean_loss < self.best_mean_loss:
             self.best_mean_loss = self.mean_loss
             shutil.copy(
-                osp.join(self.out, "checkpoint_lastest.pth.tar"),
+                osp.join(self.out, "checkpoint_latest.pth.tar"),
                 osp.join(self.out, "checkpoint_best.pth.tar"),
             )
 

diff --git a/train.py b/train.py
@@ -109,7 +109,7 @@ def main():
     # print("epoch_size (valid):", len(val_loader))
 
     if resume_from:
-        checkpoint = torch.load(osp.join(resume_from, "checkpoint_lastest.pth.tar"))
+        checkpoint = torch.load(osp.join(resume_from, "checkpoint_latest.pth.tar"))
 
     # 2. model
     if M.backbone == "stacked_hourglass":