From 38ad48c419da6be2eff9c469f21086f42e171a03 Mon Sep 17 00:00:00 2001
From: Tian Zhen <1204216974@qq.com>
Date: Mon, 25 Oct 2021 18:26:00 +0800
Subject: [PATCH] FIX: fix issue #1015

---
 recbole/trainer/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/recbole/trainer/trainer.py b/recbole/trainer/trainer.py
index 10238d3cb..5caa2c3fe 100644
--- 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']