Skip to content

Commit 41b5e4e

Browse files
committed
[update] fix param override issue
1 parent e4ba961 commit 41b5e4e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

IMDLBenCo/training_scripts/train.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,9 @@ def main(args, model_args):
304304
else:
305305
model_init_params = inspect.signature(model.__init__).parameters
306306
combined_args = {k: v for k, v in vars(args).items() if k in model_init_params}
307-
combined_args.update({k: v for k, v in vars(model_args).items() if k in model_init_params})
307+
for k, v in vars(model_args).items():
308+
if k in model_init_params and k not in combined_args:
309+
combined_args[k] = v
308310
model = model(**combined_args)
309311
# ============================================
310312
evaluator_list = [

0 commit comments

Comments
 (0)