Browse Source
update test.py
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
training
1 changed files with
3 additions and
3 deletions
-
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')\ |
|
|
|
|
