des
commited on
Upload folder using huggingface_hub
Browse files- config.json +10 -2
- model.safetensors +2 -2
- train_config.json +13 -5
config.json
CHANGED
|
@@ -13,7 +13,7 @@
|
|
| 13 |
6
|
| 14 |
]
|
| 15 |
},
|
| 16 |
-
"observation.images.
|
| 17 |
"type": "VISUAL",
|
| 18 |
"shape": [
|
| 19 |
3,
|
|
@@ -21,7 +21,15 @@
|
|
| 21 |
640
|
| 22 |
]
|
| 23 |
},
|
| 24 |
-
"observation.images.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 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/
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
|
@@ -80,7 +80,7 @@
|
|
| 80 |
6
|
| 81 |
]
|
| 82 |
},
|
| 83 |
-
"observation.images.
|
| 84 |
"type": "VISUAL",
|
| 85 |
"shape": [
|
| 86 |
3,
|
|
@@ -88,7 +88,15 @@
|
|
| 88 |
640
|
| 89 |
]
|
| 90 |
},
|
| 91 |
-
"observation.images.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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": "
|
| 134 |
"resume": false,
|
| 135 |
"seed": 1000,
|
| 136 |
-
"num_workers":
|
| 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,
|