fix extra args

Former-commit-id: 2c98a1bc3d885170f8298872c2ea2e24427fb447
This commit is contained in:
hiyouga 2024-11-09 00:24:27 +08:00
parent 91b56b7baf
commit 0027f46ccc

View File

@ -154,8 +154,8 @@ class Runner:
plot_loss=True,
ddp_timeout=180000000,
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
if get("top.checkpoint_path"):