Commit 983cc311 authored by mohammad's avatar mohammad
Browse files

further refactor, tested, and changed master to main

parent 22fa9bac
...@@ -255,7 +255,7 @@ def finetune(train_valid_datasets_provider, model_provider, ...@@ -255,7 +255,7 @@ def finetune(train_valid_datasets_provider, model_provider,
_ = load_checkpoint(model, None, None) _ = load_checkpoint(model, None, None)
args.load = original_load args.load = original_load
# This is critical when only model is loaded. We should make sure # This is critical when only model is loaded. We should make sure
# master parameters are also updated. # main parameters are also updated.
optimizer.reload_model_params() optimizer.reload_model_params()
timers('pretrained checkpoint').stop() timers('pretrained checkpoint').stop()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment