diff --git a/users/vieting/experiments/librispeech/librispeech_960_pretraining/wav2vec2/config_02_fairseq_phoneme.py b/users/vieting/experiments/librispeech/librispeech_960_pretraining/wav2vec2/config_02_fairseq_phoneme.py index 8fc0d85f5..c71bbb5c6 100644 --- a/users/vieting/experiments/librispeech/librispeech_960_pretraining/wav2vec2/config_02_fairseq_phoneme.py +++ b/users/vieting/experiments/librispeech/librispeech_960_pretraining/wav2vec2/config_02_fairseq_phoneme.py @@ -134,7 +134,7 @@ def run_fairseq_pretraining_from_checkpoint(exp_name, commit, checkpoint, python # generate config fairseq_args = get_fairseq_args(num_gpus=num_gpus) fairseq_args["task"]["alignment"] = alignment - fairseq_args["checkpoint"]["restore_file"] = checkpoint + fairseq_args["checkpoint"]["continue_once"] = checkpoint for k, v in kwargs.items(): fairseq_args["model"][k] = v