Upload folder using huggingface_hub
Browse files- config.json +8 -2
- config.yaml +19 -7
- 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.
|
|
|
9 |
"observation.images.webcam": "mean_std",
|
10 |
"observation.state": "mean_std"
|
11 |
},
|
12 |
"input_shapes": {
|
13 |
-
"observation.images.
|
|
|
|
|
|
|
|
|
|
|
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:
|
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.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
188 |
mean:
|
189 |
- - - 0.485
|
190 |
- - - 0.456
|
@@ -203,7 +210,11 @@ policy:
|
|
203 |
- 3
|
204 |
- 480
|
205 |
- 640
|
206 |
-
observation.images.
|
|
|
|
|
|
|
|
|
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.
|
|
|
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:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bcfe67d2c827beab3296442201784509193ed6fea4d82b638ca6c9375f86a536
|
3 |
+
size 206701352
|