admin/any4lerobot
1
0
Fork 0
mirror of https://github.com/Tavish9/any4lerobot.git synced 2026-05-25 18:49:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix special dataset name; make sure dataset_name is the same as oxe

Browse Source
This commit is contained in:
Tavish
2025-02-21 13:25:43 +08:00
parent 6ab4ac3202
commit 36cf315bef
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
openx_rlds.py
+1 -1
View File
@@ -81,7 +81,7 @@ def transform_raw_dataset(episode, dataset_name):
def generate_features_from_raw(builder: tfds.core.DatasetBuilder, use_videos: bool = True): def generate_features_from_raw(builder: tfds.core.DatasetBuilder, use_videos: bool = True):
dataset_name = builder.name dataset_name = Path(builder.data_dir).parent.name
state_names = [f"motor_{i}" for i in range(8)] state_names = [f"motor_{i}" for i in range(8)]
if dataset_name in OXE_DATASET_CONFIGS: if dataset_name in OXE_DATASET_CONFIGS: