gibbonbot commited on
Commit
6c0635c
·
verified ·
1 Parent(s): bc58531

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +78 -0
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ datasets: AgentAppStore/Test_Data_Ex
4
+ library_name: gibbonbot
5
+ pipeline_tag: robotics
6
+ model_name: pi0.5
7
+ tags:
8
+ - gibbonbot
9
+ - pi0.5
10
+ task_categories:
11
+ - robotics
12
+ ---
13
+
14
+ # pi0.5 model - 🧪 gibbonbot training pipeline
15
+
16
+ - **Dataset**: [AgentAppStore/Test_Data_Ex](https://huggingface.co/datasets/AgentAppStore/Test_Data_Ex)
17
+ - **Wandb run id**: None
18
+
19
+
20
+ ## Error Traceback
21
+ We faced an issue while training your model.
22
+
23
+ ```
24
+ Caught KeyError in DataLoader worker process 0.
25
+ Original Traceback (most recent call last):
26
+ File "/app/.venv/lib/python3.11/site-packages/torch/utils/data/_utils/worker.py", line 349, in _worker_loop
27
+ data = fetcher.fetch(index) # type: ignore[possibly-undefined]
28
+ ^^^^^^^^^^^^^^^^^^^^
29
+ File "/app/.venv/lib/python3.11/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
30
+ data = [self.dataset[idx] for idx in possibly_batched_index]
31
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
32
+ File "/app/.venv/lib/python3.11/site-packages/torch/utils/data/_utils/fetch.py", line 52, in <listcomp>
33
+ data = [self.dataset[idx] for idx in possibly_batched_index]
34
+ ~~~~~~~~~~~~^^^^^
35
+ File "/openpi-source/src/openpi/training/data_loader.py", line 58, in __getitem__
36
+ return self._transform(self._dataset[index])
37
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
38
+ File "/openpi-source/src/openpi/transforms.py", line 70, in __call__
39
+ data = transform(data)
40
+ ^^^^^^^^^^^^^^^
41
+ File "/openpi-source/src/openpi/transforms.py", line 101, in __call__
42
+ return jax.tree.map(lambda k: flat_item[k], self.structure)
43
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
44
+ File "/app/.venv/lib/python3.11/site-packages/jax/_src/tree.py", line 155, in map
45
+ return tree_util.tree_map(f, tree, *rest, is_leaf=is_leaf)
46
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
47
+ File "/app/.venv/lib/python3.11/site-packages/jax/_src/tree_util.py", line 358, in tree_map
48
+ return treedef.unflatten(f(*xs) for xs in zip(*all_leaves))
49
+ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
50
+ File "/app/.venv/lib/python3.11/site-packages/jax/_src/tree_util.py", line 358, in <genexpr>
51
+ return treedef.unflatten(f(*xs) for xs in zip(*all_leaves))
52
+ ^^^^^^
53
+ File "/openpi-source/src/openpi/transforms.py", line 101, in <lambda>
54
+ return jax.tree.map(lambda k: flat_item[k], self.structure)
55
+ ~~~~~~~~~^^^
56
+ KeyError: 'observation.images.secondary_0'
57
+
58
+ ```
59
+
60
+
61
+ ## Training parameters
62
+
63
+ ```text
64
+ {
65
+ "save_interval": 100,
66
+ "num_train_steps": 500,
67
+ "batch_size": 32,
68
+ "seed": 42,
69
+ "data.image_keys": [
70
+ "observation.images.main",
71
+ "observation.images.secondary_0"
72
+ ]
73
+ }
74
+ ```
75
+
76
+ 📖 **Get Started**: [docs.gibbon.bot](https://docs.gibbon.bot?utm_source=huggingface_readme)
77
+
78
+ 🤖 **Get your robot**: [robots.gibbon.bot](https://robots.gibbon.bot?utm_source=huggingface_readme)