brandonyang commited on
Commit
c6b9fbf
·
verified ·
1 Parent(s): 166aee0

Upload policy

Browse files
Files changed (2) hide show
  1. config.json +32 -2
  2. model.safetensors +2 -2
config.json CHANGED
@@ -7,8 +7,38 @@
7
  "STATE": "MEAN_STD",
8
  "ACTION": "MEAN_STD"
9
  },
10
- "input_features": {},
11
- "output_features": {},
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  "device": "cuda",
13
  "use_amp": false,
14
  "push_to_hub": true,
 
7
  "STATE": "MEAN_STD",
8
  "ACTION": "MEAN_STD"
9
  },
10
+ "input_features": {
11
+ "observation.state": {
12
+ "type": "STATE",
13
+ "shape": [
14
+ 8
15
+ ]
16
+ },
17
+ "observation.images.wrist_cam": {
18
+ "type": "VISUAL",
19
+ "shape": [
20
+ 3,
21
+ 224,
22
+ 224
23
+ ]
24
+ },
25
+ "observation.images.external_cam": {
26
+ "type": "VISUAL",
27
+ "shape": [
28
+ 3,
29
+ 224,
30
+ 224
31
+ ]
32
+ }
33
+ },
34
+ "output_features": {
35
+ "action": {
36
+ "type": "ACTION",
37
+ "shape": [
38
+ 8
39
+ ]
40
+ }
41
+ },
42
  "device": "cuda",
43
  "use_amp": false,
44
  "push_to_hub": true,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6db4cba905150a2ee0164bb32dd933167075ec86ba52db6be2ac35a7261d6540
3
- size 12952325440
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d6a7a41c7df330741c4f4c60c640a4a737f4d37017681fdbafe609ec690c41a
3
+ size 12952326304