gibbonbot/pi0.5-Test_Data_Ex-hmuwqlvkrt

pi0.5 model - 🧪 gibbonbot training pipeline

• Dataset: AgentAppStore/Test_Data_Ex
• Wandb run id: None

Error Traceback

We faced an issue while training your model.

Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/app/.venv/lib/python3.11/site-packages/torch/utils/data/_utils/worker.py", line 349, in _worker_loop
    data = fetcher.fetch(index)  # type: ignore[possibly-undefined]
           ^^^^^^^^^^^^^^^^^^^^
  File "/app/.venv/lib/python3.11/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/app/.venv/lib/python3.11/site-packages/torch/utils/data/_utils/fetch.py", line 52, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
            ~~~~~~~~~~~~^^^^^
  File "/openpi-source/src/openpi/training/data_loader.py", line 58, in __getitem__
    return self._transform(self._dataset[index])
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/openpi-source/src/openpi/transforms.py", line 70, in __call__
    data = transform(data)
           ^^^^^^^^^^^^^^^
  File "/openpi-source/src/openpi/transforms.py", line 263, in __call__
    prompt = prompt.item()
             ^^^^^^^^^^^
AttributeError: 'int' object has no attribute 'item'

Training parameters

{
  "save_interval": 100,
  "num_train_steps": 500,
  "batch_size": 32,
  "seed": 42,
  "data.image_keys": [
    "observation.images.main"
  ]
}

📖 Get Started: docs.gibbon.bot

🤖 Get your robot: robots.gibbon.bot