admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-17 17:50:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(rebase) reverting files to main

Browse Source
This commit is contained in:
Michel Aractingi
2025-06-30 16:22:57 +02:00
parent c3e98db37d
commit d9b9cc80da
13 changed files with 339 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/artifacts/policies/save_policy_to_safetensors.py
+1 -1
View File
@@ -32,7 +32,7 @@ def get_policy_stats(ds_repo_id: str, policy_name: str, policy_kwargs: dict):
train_cfg = TrainPipelineConfig(
# TODO(rcadene, aliberts): remove dataset download
dataset=DatasetConfig(repo_id=ds_repo_id, episodes=[0]),
policy=make_policy_config(policy_name, **policy_kwargs),
policy=make_policy_config(policy_name, push_to_hub=False, **policy_kwargs),
)
train_cfg.validate() # Needed for auto-setting some parameters