gibbonbot/pi0.5-Test_Data_Ex-333bkfqsow

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 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

{
  "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

🤖 Get your robot: robots.gibbon.bot