diff --git a/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py b/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py index 9db26c7c..7e0ce172 100644 --- a/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py +++ b/lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py @@ -33,7 +33,7 @@ from lerobot.common.utils.utils import init_logging def check_format(raw_dir) -> bool: # TODO(rcadene): remove hardcoding - raw_dir = raw_dir / "018f9fdc-6b7b-7432-a529-40d2cc718032" + raw_dir = raw_dir / "018fa076-ad19-7c77-afa4-49f7f072e86f" assert raw_dir.exists() leader_file = list(raw_dir.glob("*.parquet")) @@ -44,7 +44,7 @@ def check_format(raw_dir) -> bool: def load_from_raw(raw_dir: Path, out_dir: Path): # TODO(rcadene): remove hardcoding - raw_dir = raw_dir / "018f9fdc-6b7b-7432-a529-40d2cc718032" + raw_dir = raw_dir / "018fa076-ad19-7c77-afa4-49f7f072e86f" # Load data stream that will be used as reference for the timestamps synchronization reference_key = "observation.images.cam_right_wrist" @@ -91,7 +91,7 @@ def load_from_raw(raw_dir: Path, out_dir: Path): # sanity check episode indices go from 0 to n-1 ep_ids = [ep_idx for ep_idx, _ in df.groupby("episode_index")] - expected_ep_ids = list(range(df["episode_index"].max())) + expected_ep_ids = list(range(df["episode_index"].max() + 1)) assert ep_ids == expected_ep_ids, f"Episodes indices go from {ep_ids} instead of {expected_ep_ids}" # Create symlink to raw videos directory (that needs to be absolute not relative) diff --git a/lerobot/configs/policy/act_real_world.yaml b/lerobot/configs/policy/act_real_world.yaml index b2e293ee..c16da75e 100644 --- a/lerobot/configs/policy/act_real_world.yaml +++ b/lerobot/configs/policy/act_real_world.yaml @@ -25,8 +25,8 @@ training: offline_steps: 80000 online_steps: 0 eval_freq: 99999999999999 - save_freq: 100000 - log_freq: 250 + save_freq: 1000 + log_freq: 100 save_model: true batch_size: 8