Yichao Zhou 7 лет назад
Родитель
Сommit
437ef2922e
2 измененных файлов с 4 добавлено и 4 удалено
  1. 3 3
      lcnn/trainer.py
  2. 1 1
      train.py

+ 3 - 3
lcnn/trainer.py

@@ -144,16 +144,16 @@ class Trainer(object):
                 "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"),
             )
 

+ 1 - 1
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":