ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-04 21:01:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

smaller model

Browse Source
This commit is contained in:
Pepijn
2025-08-28 17:43:03 +02:00
parent c877e98658
commit bead25a58a
10 changed files with 235 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/lerobot/scripts/train.py
View File

@@ -137,7 +137,7 @@ def train(cfg: TrainPipelineConfig):
logging.info("Creating policy")
# Pass episode_data_index for RLearN to calculate proper progress
episode_data_index = dataset.episode_data_index if hasattr(dataset, 'episode_data_index') else None
episode_data_index = dataset.episode_data_index if hasattr(dataset, "episode_data_index") else None
policy = make_policy(
cfg=cfg.policy,
ds_meta=dataset.meta,