diff --git a/ppcls/engine/train/train.py b/ppcls/engine/train/train.py
index 3b02bac8f305d9b944eb97a5a98ba3c39521191a..b15c1088ae23b16a1e1c724a5c783f75cb5449ce 100644
--- a/ppcls/engine/train/train.py
+++ b/ppcls/engine/train/train.py
@@ -21,7 +21,6 @@ from ppcls.utils import profiler
 
 def train_epoch(engine, epoch_id, print_batch_step):
     tic = time.time()
-    v_current = [int(i) for i in paddle.__version__.split(".")]
     for iter_id, batch in enumerate(engine.train_dataloader):
         if iter_id >= engine.max_iter:
             break