diff --git a/tools/program.py b/tools/program.py
index a12e336086f7a13ec2174e05afa23cf26d02b9d5..67cef9bc259a18bf0ca514c95e731506302ed493 100755
--- a/tools/program.py
+++ b/tools/program.py
@@ -348,7 +348,7 @@ def train_eval_rec_run(config, exe, train_info_dict, eval_info_dict):
 
         except fluid.core.EOFException:
             train_loader.reset()
-        if epoch ==0 and save_epoch_step == 1:
+        if epoch == 0 and save_epoch_step == 1:
             save_path = save_model_dir + "/iter_epoch_0"
             save_model(train_info_dict['train_program'],save_path)
         if epoch > 0 and epoch % save_epoch_step == 0: