🐛 bug fixes

2026-03-11 23:07:37 +01:00
parent 4115447022
commit 1a822bd82e
4 changed files with 34 additions and 7 deletions
--- a/scripts/train.py
+++ b/scripts/train.py
@@ -101,6 +101,12 @@ def main(cfg: DictConfig) -> None:
    training_dict.pop("hpo", None)  # HPO range metadata — not a TrainerConfig field
    task = _init_clearml(choices, remote=remote)

+    # Drop keys not recognised by TrainerConfig (e.g. ClearML-injected
+    # resume_from_task_id or any future additions)
+    import dataclasses as _dc
+    _valid_keys = {f.name for f in _dc.fields(TrainerConfig)}
+    training_dict = {k: v for k, v in training_dict.items() if k in _valid_keys}
+
    env_name = choices.get("env", "cartpole")
    env = build_env(env_name, cfg)
    runner = _build_runner(choices.get("runner", "mujoco"), env, cfg)