---
datasets: AgentAppStore/Test_Data_Ex
library_name: gibbonbot
pipeline_tag: robotics
model_name: pi0.5
tags:
- gibbonbot
- pi0.5
task_categories:
- robotics
---

# pi0.5 model - 🧪 gibbonbot training pipeline

- **Dataset**: [AgentAppStore/Test_Data_Ex](https://huggingface.co/datasets/AgentAppStore/Test_Data_Ex)
- **Wandb run id**: None


## Error Traceback
We faced an issue while training your model.

```
Caught KeyError 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 101, in __call__
    return jax.tree.map(lambda k: flat_item[k], self.structure)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/app/.venv/lib/python3.11/site-packages/jax/_src/tree.py", line 155, in map
    return tree_util.tree_map(f, tree, *rest, is_leaf=is_leaf)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/app/.venv/lib/python3.11/site-packages/jax/_src/tree_util.py", line 358, in tree_map
    return treedef.unflatten(f(*xs) for xs in zip(*all_leaves))
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/app/.venv/lib/python3.11/site-packages/jax/_src/tree_util.py", line 358, in <genexpr>
    return treedef.unflatten(f(*xs) for xs in zip(*all_leaves))
                             ^^^^^^
  File "/openpi-source/src/openpi/transforms.py", line 101, in <lambda>
    return jax.tree.map(lambda k: flat_item[k], self.structure)
                                  ~~~~~~~~~^^^
KeyError: 'observation.images.secondary_0'

```


## Training parameters

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

📖 **Get Started**: [docs.gibbon.bot](https://docs.gibbon.bot?utm_source=huggingface_readme)

🤖 **Get your robot**: [robots.gibbon.bot](https://robots.gibbon.bot?utm_source=huggingface_readme)