Hugging Face's logo

lerobot
/
pi05_base

Safetensors
English
Model card Files
xet
Community
4
pepijn223 HF Staff commited on
Commit
8772304
·
verified ·
1 Parent(s): 92e5004

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +40 -2
config.json CHANGED
@@ -6,8 +6,46 @@
6
  "STATE": "MEAN_STD",
7
  "ACTION": "MEAN_STD"
8
  },
9
- "input_features": {},
10
- "output_features": {},
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  "device": "mps",
12
  "use_amp": false,
13
  "push_to_hub": true,
 
6
  "STATE": "MEAN_STD",
7
  "ACTION": "MEAN_STD"
8
  },
9
+ "input_features": {
10
+ "observation.images.base_0_rgb": {
11
+ "type": "VISUAL",
12
+ "shape": [
13
+ 3,
14
+ 224,
15
+ 224
16
+ ]
17
+ },
18
+ "observation.images.left_wrist_0_rgb": {
19
+ "type": "VISUAL",
20
+ "shape": [
21
+ 3,
22
+ 224,
23
+ 224
24
+ ]
25
+ },
26
+ "observation.images.right_wrist_0_rgb": {
27
+ "type": "VISUAL",
28
+ "shape": [
29
+ 3,
30
+ 224,
31
+ 224
32
+ ]
33
+ },
34
+ "observation.state": {
35
+ "type": "STATE",
36
+ "shape": [
37
+ 8
38
+ ]
39
+ }
40
+ },
41
+ "output_features": {
42
+ "action": {
43
+ "type": "ACTION",
44
+ "shape": [
45
+ 7
46
+ ]
47
+ }
48
+ },
49
  "device": "mps",
50
  "use_amp": false,
51
  "push_to_hub": true,