diff --git a/lcnn/trainer.py b/lcnn/trainer.py index 748eec3..872528b 100644 --- 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 index cb35dc7..8f95196 100755 --- 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":