Merge remote-tracking branch 'origin/main' into user/rcadene/2025_04_11_dataset_v3

2026-05-20 11:09:59 +00:00 · 2025-04-21 11:03:12 +02:00
parent 2866d0770f 483be9aac2
commit 38c7ac5b07
287 changed files with 31690 additions and 12428 deletions
@@ -283,6 +283,9 @@ def train(cfg: TrainPipelineConfig):
        eval_env.close()
    logging.info("End of training")

+    if cfg.policy.push_to_hub:
+        policy.push_model_to_hub(cfg)
+

 if __name__ == "__main__":
    init_logging()