Skip to content

Commit

Permalink
use fairseq param for start checkpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
AndreasPlt committed Dec 3, 2024
1 parent 3464e55 commit 2845bdd
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 2845bdd

Please sign in to comment.