diff --git a/paddlehub/finetune/task.py b/paddlehub/finetune/task.py
index 30e3f72f1c9e84c9053b88f9db5cdb877c376a64..45f36cf256d74906c5a11849fdb6d0d4ed1307d1 100644
--- a/paddlehub/finetune/task.py
+++ b/paddlehub/finetune/task.py
@@ -443,7 +443,7 @@ class BasicTask(object):
             "step %d: [step/sec: %.2f]" % (self.current_step, run_speed))
 
     def _save_ckpt_interval_event(self):
-        self.save_checkpoint(self.current_epoch, self.current_step)
+        self.save_checkpoint()
 
     def _eval_interval_event(self):
         self.eval(phase="dev")
@@ -469,7 +469,7 @@ class BasicTask(object):
 
     # NOTE: current saved checkpoint machanism is not completed,
     # it can't restore dataset training status
-    def save_checkpoint(self, epoch, step):
+    def save_checkpoint(self):
         save_checkpoint(
             checkpoint_dir=self.config.checkpoint_dir,
             current_epoch=self.current_epoch,
@@ -512,7 +512,7 @@ class BasicTask(object):
                     self.env.current_epoch += 1
 
                 # Save checkpoint after finetune
-                self.save_checkpoint(self.current_epoch + 1, self.current_step)
+                self.save_checkpoint()
 
                 # Final evaluation
                 self.eval(phase="dev")