Hugging Face's logo

fracapuano
/
pi0

Robotics
LeRobot
Safetensors
Model card Files
xet
Community
fracapuano HF Staff commited on
Commit
351a6f7
·
verified ·
1 Parent(s): 13027c8

fixing pi0's config

Browse files
Files changed (1) hide show
  1. config.json +40 -2
config.json CHANGED
@@ -8,8 +8,46 @@
8
  "STATE": "MEAN_STD",
9
  "ACTION": "MEAN_STD"
10
  },
11
- "input_features": {},
12
- "output_features": {},
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  "chunk_size": 50,
14
  "n_action_steps": 50,
15
  "max_state_dim": 32,
 
8
  "STATE": "MEAN_STD",
9
  "ACTION": "MEAN_STD"
10
  },
11
+ "input_features": {
12
+ "observation.state": {
13
+ "type": "STATE",
14
+ "shape": [
15
+ 6
16
+ ]
17
+ },
18
+ "observation.image2": {
19
+ "type": "VISUAL",
20
+ "shape": [
21
+ 3,
22
+ 256,
23
+ 256
24
+ ]
25
+ },
26
+ "observation.image": {
27
+ "type": "VISUAL",
28
+ "shape": [
29
+ 3,
30
+ 256,
31
+ 256
32
+ ]
33
+ },
34
+ "observation.image3": {
35
+ "type": "VISUAL",
36
+ "shape": [
37
+ 3,
38
+ 256,
39
+ 256
40
+ ]
41
+ }
42
+ },
43
+ "output_features": {
44
+ "action": {
45
+ "type": "ACTION",
46
+ "shape": [
47
+ 6
48
+ ]
49
+ }
50
+ },
51
  "chunk_size": 50,
52
  "n_action_steps": 50,
53
  "max_state_dim": 32,