From 03ac76d8fcb87b0b3afcc58ff907533d78f05f87 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 25 Feb 2022 10:10:47 +0800 Subject: [PATCH] update test.py Signed-off-by: Jael Gu --- test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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')\