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

add decode logging

Browse Source
This commit is contained in:
Pepijn
2025-08-30 16:08:21 +02:00
parent b357a8c4d8
commit b1ff7132c1
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/train.py
+1 -7
View File
@@ -295,13 +295,7 @@ def train(cfg: TrainPipelineConfig):
torch.backends.cuda.matmul.allow_tf32 = True torch.backends.cuda.matmul.allow_tf32 = True
logging.info("Creating dataset") logging.info("Creating dataset")
# Pass video backend to dataset for RLearN optimization dataset = make_dataset(cfg)
dataset_kwargs = {}
if getattr(cfg.policy, "type", None) == "rlearn" and hasattr(cfg.policy, "video_backend"):
dataset_kwargs["video_backend"] = cfg.policy.video_backend
logging.info(f"Using video backend: {cfg.policy.video_backend}")
dataset = make_dataset(cfg, **dataset_kwargs)
# Add video decoding timing for RLearN debugging # Add video decoding timing for RLearN debugging
if getattr(cfg.policy, "type", None) == "rlearn": if getattr(cfg.policy, "type", None) == "rlearn":