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
|