hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

fix(codec): hot-fix for default codec in linux arm platforms (#868)

Browse Source
This commit is contained in:
Steven Palma
2025-03-17 13:23:11 +01:00
committed by GitHub
parent 9f0a8a49d0
commit 1c15bab70f
5 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/configs/default.py
View File

@@ -20,6 +20,7 @@ from lerobot.common import (
policies, # noqa: F401
)
from lerobot.common.datasets.transforms import ImageTransformsConfig
from lerobot.common.datasets.video_utils import get_safe_default_codec
@dataclass
@@ -35,7 +36,7 @@ class DatasetConfig:
image_transforms: ImageTransformsConfig = field(default_factory=ImageTransformsConfig)
revision: str | None = None
use_imagenet_stats: bool = True
video_backend: str = "pyav"
video_backend: str = field(default_factory=get_safe_default_codec)
@dataclass