fix extra args

This commit is contained in:
hiyouga
2024-11-09 00:24:27 +08:00
parent 707ff5a370
commit eb294fed13

View File

@@ -154,8 +154,8 @@ class Runner:
plot_loss=True, plot_loss=True,
ddp_timeout=180000000, ddp_timeout=180000000,
include_num_input_tokens_seen=False if is_transformers_version_equal_to_4_46() else True, # FIXME include_num_input_tokens_seen=False if is_transformers_version_equal_to_4_46() else True, # FIXME
**json.loads(get("train.extra_args")),
) )
args.update(json.loads(get("train.extra_args")))
# checkpoints # checkpoints
if get("top.checkpoint_path"): if get("top.checkpoint_path"):