des commited on
Commit
e2c7c7d
·
verified ·
1 Parent(s): 410c6be

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. config.json +10 -2
  2. model.safetensors +2 -2
  3. train_config.json +13 -5
config.json CHANGED
@@ -13,7 +13,7 @@
13
  6
14
  ]
15
  },
16
- "observation.images.laptop": {
17
  "type": "VISUAL",
18
  "shape": [
19
  3,
@@ -21,7 +21,15 @@
21
  640
22
  ]
23
  },
24
- "observation.images.phone": {
 
 
 
 
 
 
 
 
25
  "type": "VISUAL",
26
  "shape": [
27
  3,
 
13
  6
14
  ]
15
  },
16
+ "observation.images.top": {
17
  "type": "VISUAL",
18
  "shape": [
19
  3,
 
21
  640
22
  ]
23
  },
24
+ "observation.images.front": {
25
+ "type": "VISUAL",
26
+ "shape": [
27
+ 3,
28
+ 480,
29
+ 640
30
+ ]
31
+ },
32
+ "observation.images.wrist": {
33
  "type": "VISUAL",
34
  "shape": [
35
  3,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d318f2435b282d5efa278931e2f084de6829c4f9ff1ec5f5956283f62b09f593
3
- size 206701072
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a2d6bafa968d23aa616e18187ab88d6867e8ba6b71e111cc61cce0d7add47b0
3
+ size 206701336
train_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "dataset": {
3
- "repo_id": "moveit2/so101_test",
4
  "root": null,
5
  "episodes": null,
6
  "image_transforms": {
@@ -80,7 +80,7 @@
80
  6
81
  ]
82
  },
83
- "observation.images.laptop": {
84
  "type": "VISUAL",
85
  "shape": [
86
  3,
@@ -88,7 +88,15 @@
88
  640
89
  ]
90
  },
91
- "observation.images.phone": {
 
 
 
 
 
 
 
 
92
  "type": "VISUAL",
93
  "shape": [
94
  3,
@@ -130,10 +138,10 @@
130
  "optimizer_lr_backbone": 1e-05
131
  },
132
  "output_dir": "outputs/train/act_so101_test",
133
- "job_name": "act_so101_test10",
134
  "resume": false,
135
  "seed": 1000,
136
- "num_workers": 4,
137
  "batch_size": 8,
138
  "steps": 100000,
139
  "eval_freq": 20000,
 
1
  {
2
  "dataset": {
3
+ "repo_id": "moveit2/so101_test2",
4
  "root": null,
5
  "episodes": null,
6
  "image_transforms": {
 
80
  6
81
  ]
82
  },
83
+ "observation.images.top": {
84
  "type": "VISUAL",
85
  "shape": [
86
  3,
 
88
  640
89
  ]
90
  },
91
+ "observation.images.front": {
92
+ "type": "VISUAL",
93
+ "shape": [
94
+ 3,
95
+ 480,
96
+ 640
97
+ ]
98
+ },
99
+ "observation.images.wrist": {
100
  "type": "VISUAL",
101
  "shape": [
102
  3,
 
138
  "optimizer_lr_backbone": 1e-05
139
  },
140
  "output_dir": "outputs/train/act_so101_test",
141
+ "job_name": "act_so101_test",
142
  "resume": false,
143
  "seed": 1000,
144
+ "num_workers": 0,
145
  "batch_size": 8,
146
  "steps": 100000,
147
  "eval_freq": 20000,