Merge pull request #1021 from chenyuwuxin/master

FIX: fix issue #1015
RUCAIBox · Oct 26, 2021 · deb1f30 · deb1f30
2 parents 186a9f9 + f3c8354
commit deb1f30
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/recbole/trainer/trainer.py b/recbole/trainer/trainer.py
@@ -223,6 +223,7 @@ def resume_checkpoint(self, resume_file):
 
         """
         resume_file = str(resume_file)
+        self.saved_model_file = resume_file
         checkpoint = torch.load(resume_file)
         self.start_epoch = checkpoint['epoch'] + 1
         self.cur_step = checkpoint['cur_step']