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

further refactor, tested, and changed master to main

parent 22fa9bac

tasks/finetune_utils.py

+1 −1
Original line number Diff line number Diff line
@@ -255,7 +255,7 @@ def finetune(train_valid_datasets_provider, model_provider, 
        _ = load_checkpoint(model, None, None)

        args.load = original_load

        # 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()

    timers('pretrained checkpoint').stop()