gr00t model - 🧪 gibbonbot training pipeline
- Dataset: AgentAppStore/Test_Data_Ex
- Wandb run id: None
Error Traceback
We faced an issue while training your model.
Traceback (most recent call last):
File "/root/src/helper.py", line 140, in train_gr00t_on_modal
trainer.train(
File "/root/gibbonbot/am/gr00t.py", line 1247, in train
asyncio.run(
File "/usr/local/lib/python3.11/asyncio/runners.py", line 190, in run
return runner.run(main)
^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/asyncio/runners.py", line 118, in run
return self._loop.run_until_complete(task)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/asyncio/base_events.py", line 654, in run_until_complete
return future.result()
^^^^^^^^^^^^^^^
File "/root/gibbonbot/am/gr00t.py", line 1459, in _call_training_script
raise RuntimeError(error_msg)
RuntimeError: Training process failed with exit code 1:
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/gr00t/gr00t/data/dataset.py", line 566, in get_step_data
self.curr_traj_data = self.get_trajectory_data(trajectory_id)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/gr00t/gr00t/data/dataset.py", line 579, in get_trajectory_data
parquet_path = self.dataset_path / self.data_path_pattern.format(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
KeyError: 'chunk_index'
0%| | 0/80 [00:01<?, ?it/s]
Training parameters
{
"validation_dataset_name": null,
"batch-size": 107,
"num-epochs": 10,
"save-steps": 800,
"learning_rate": 0.0001,
"data_dir": "/tmp/outputs/data",
"validation_data_dir": "/tmp/outputs/validation_data",
"output_dir": "/tmp/outputs/train"
}
📖 Get Started: docs.gibbon.bot
🤖 Get your robot: robots.gibbon.bot