gr00t Model - phospho Training Pipeline

Error Traceback

We faced an issue while training your model.

Traceback (most recent call last):
  File "/root/src/helper.py", line 353, in predict
    raise RuntimeError(error_msg)
RuntimeError: Training process failed with exit code 1:
File "/workspace/gr00t/data/dataset.py", line 526, in get_step_data
self.curr_traj_data = self.get_trajectory_data(trajectory_id)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/gr00t/data/dataset.py", line 542, in get_trajectory_data
assert parquet_path.exists(), f"Parquet file not found at {parquet_path}"
^^^^^^^^^^^^^^^^^^^^^
AssertionError: Parquet file not found at /tmp/outputs/data/data/chunk-000/episode_000000.parquet


0%|          | 0/1750 [00:05<?, ?it/s]

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/root/src/helper.py", line 355, in predict
    raise RuntimeError(e)
RuntimeError: Training process failed with exit code 1:
File "/workspace/gr00t/data/dataset.py", line 526, in get_step_data
self.curr_traj_data = self.get_trajectory_data(trajectory_id)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/gr00t/data/dataset.py", line 542, in get_trajectory_data
assert parquet_path.exists(), f"Parquet file not found at {parquet_path}"
^^^^^^^^^^^^^^^^^^^^^
AssertionError: Parquet file not found at /tmp/outputs/data/data/chunk-000/episode_000000.parquet


0%|          | 0/1750 [00:05<?, ?it/s]

Training parameters:

Dataset: nebo1337/GetTheRubberNextG2
Wandb run URL: None
Epochs: 10
Batch size: 64
Training steps: 1743

📖 Get Started: docs.phospho.ai

🤖 Get your robot: robots.phospho.ai