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 165, in predict
    trainer.train(timeout_seconds=timeout_seconds)
  File "/root/phosphobot/am/gr00t.py", line 1146, in train
    asyncio.run(
  File "/opt/conda/lib/python3.11/asyncio/runners.py", line 190, in run
    return runner.run(main)
           ^^^^^^^^^^^^^^^^
  File "/opt/conda/lib/python3.11/asyncio/runners.py", line 118, in run
    return self._loop.run_until_complete(task)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/lib/python3.11/asyncio/base_events.py", line 654, in run_until_complete
    return future.result()
           ^^^^^^^^^^^^^^^
  File "/root/phosphobot/am/gr00t.py", line 996, in run_gr00t_training
    raise RuntimeError(error_msg)
RuntimeError: Training process failed with exit code 1:
File "/opt/conda/lib/python3.11/site-packages/pandas/core/indexing.py", line 1558, in _get_listlike_indexer
keyarr, indexer = ax._get_indexer_strict(key, axis_name)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/conda/lib/python3.11/site-packages/pandas/core/indexes/base.py", line 6200, in _get_indexer_strict
self._raise_if_missing(keyarr, indexer, axis_name)
File "/opt/conda/lib/python3.11/site-packages/pandas/core/indexes/base.py", line 6249, in _raise_if_missing
raise KeyError(f"None of [{key}] are in the [{axis_name}]")
KeyError: "None of [Index([1], dtype='int64', name='task_index')] are in the [index]"

0%|          | 0/1690 [00:02<?, ?it/s]

Training parameters:

Dataset: Schmidie/eyes
Wandb run URL: None
Epochs: 10
Batch size: 49
Training steps: None

📖 Get Started: docs.phospho.ai

🤖 Get your robot: robots.phospho.ai