fix extra args

Former-commit-id: eb294fed1319e97657b54b9aa479d87734c711dc
This commit is contained in:
hiyouga 2024-11-09 00:24:27 +08:00
parent 265875fffd
commit d7e35ddcaa

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"):