oldTOM commited on
Commit
d73e1f3
·
verified ·
1 Parent(s): b335b43

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. config.json +8 -2
  2. config.yaml +19 -7
  3. model.safetensors +2 -2
config.json CHANGED
@@ -5,12 +5,18 @@
5
  "dropout": 0.1,
6
  "feedforward_activation": "relu",
7
  "input_normalization_modes": {
8
- "observation.images.phone": "mean_std",
 
9
  "observation.images.webcam": "mean_std",
10
  "observation.state": "mean_std"
11
  },
12
  "input_shapes": {
13
- "observation.images.phone": [
 
 
 
 
 
14
  3,
15
  480,
16
  640
 
5
  "dropout": 0.1,
6
  "feedforward_activation": "relu",
7
  "input_normalization_modes": {
8
+ "observation.images.android": "mean_std",
9
+ "observation.images.handcam": "mean_std",
10
  "observation.images.webcam": "mean_std",
11
  "observation.state": "mean_std"
12
  },
13
  "input_shapes": {
14
+ "observation.images.android": [
15
+ 3,
16
+ 480,
17
+ 640
18
+ ],
19
+ "observation.images.handcam": [
20
  3,
21
  480,
22
  640
config.yaml CHANGED
@@ -2,14 +2,12 @@ resume: false
2
  device: cuda
3
  use_amp: false
4
  seed: 1000
5
- dataset_repo_id:
6
- - underctrl/mutli-stacked-block_mutli-color_pick-up_80
7
- - underctrl/single-stacked-block_mutli-color_pick-up_80
8
  video_backend: pyav
9
  training:
10
  offline_steps: 80000
11
  num_workers: 4
12
- batch_size: 8
13
  eval_freq: -1
14
  log_freq: 100
15
  save_checkpoint: true
@@ -184,7 +182,16 @@ override_dataset_stats:
184
  - - - 0.229
185
  - - - 0.224
186
  - - - 0.225
187
- observation.images.phone:
 
 
 
 
 
 
 
 
 
188
  mean:
189
  - - - 0.485
190
  - - - 0.456
@@ -203,7 +210,11 @@ policy:
203
  - 3
204
  - 480
205
  - 640
206
- observation.images.phone:
 
 
 
 
207
  - 3
208
  - 480
209
  - 640
@@ -214,7 +225,8 @@ policy:
214
  - ${env.action_dim}
215
  input_normalization_modes:
216
  observation.images.webcam: mean_std
217
- observation.images.phone: mean_std
 
218
  observation.state: mean_std
219
  output_normalization_modes:
220
  action: mean_std
 
2
  device: cuda
3
  use_amp: false
4
  seed: 1000
5
+ dataset_repo_id: underctrl/handcamera_single-block_blue-color_pick-up_80
 
 
6
  video_backend: pyav
7
  training:
8
  offline_steps: 80000
9
  num_workers: 4
10
+ batch_size: 16
11
  eval_freq: -1
12
  log_freq: 100
13
  save_checkpoint: true
 
182
  - - - 0.229
183
  - - - 0.224
184
  - - - 0.225
185
+ observation.images.android:
186
+ mean:
187
+ - - - 0.485
188
+ - - - 0.456
189
+ - - - 0.406
190
+ std:
191
+ - - - 0.229
192
+ - - - 0.224
193
+ - - - 0.225
194
+ observation.images.handcam:
195
  mean:
196
  - - - 0.485
197
  - - - 0.456
 
210
  - 3
211
  - 480
212
  - 640
213
+ observation.images.android:
214
+ - 3
215
+ - 480
216
+ - 640
217
+ observation.images.handcam:
218
  - 3
219
  - 480
220
  - 640
 
225
  - ${env.action_dim}
226
  input_normalization_modes:
227
  observation.images.webcam: mean_std
228
+ observation.images.android: mean_std
229
+ observation.images.handcam: mean_std
230
  observation.state: mean_std
231
  output_normalization_modes:
232
  action: mean_std
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f40c4c53d07569198f41dd3e500fe222e41e113213c5f59eaae10c2d5021b1d9
3
- size 206701072
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcfe67d2c827beab3296442201784509193ed6fea4d82b638ca6c9375f86a536
3
+ size 206701352