diff --git a/test.py b/test.py
index 7fed63c..723ee8e 100644
--- a/test.py
+++ b/test.py
@@ -87,9 +87,9 @@ if __name__ == '__main__':
     freezer = LayerFreezer(op.get_model())
     freezer.set_slice(-1)
 
-    # op.train(training_config, train_dataset=train_data, eval_dataset=eval_data)
-    op.train(training_config, train_dataset=train_data, eval_dataset=eval_data,
-             resume_checkpoint_path=training_config.output_dir + '/epoch_2')
+    op.train(training_config, train_dataset=train_data, eval_dataset=eval_data)
+    # op.train(training_config, train_dataset=train_data, eval_dataset=eval_data,
+    #          resume_checkpoint_path=training_config.output_dir + '/epoch_2')
 
     # op.save('./test_save')
     # op.load('./test_save')\