Upload policy weights, train config and readme
Browse files- README.md +2 -2
- config.json +9 -1
- model.safetensors +2 -2
- train_config.json +13 -5
README.md
CHANGED
|
@@ -1,13 +1,13 @@
|
|
| 1 |
---
|
| 2 |
-
datasets: LeoMuuuu/hanoi-
|
| 3 |
library_name: lerobot
|
| 4 |
license: apache-2.0
|
| 5 |
model_name: act
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
-
- robotics
|
| 9 |
- lerobot
|
| 10 |
- act
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for act
|
|
|
|
| 1 |
---
|
| 2 |
+
datasets: LeoMuuuu/hanoi-red2middle
|
| 3 |
library_name: lerobot
|
| 4 |
license: apache-2.0
|
| 5 |
model_name: act
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
|
|
|
| 8 |
- lerobot
|
| 9 |
- act
|
| 10 |
+
- robotics
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for act
|
config.json
CHANGED
|
@@ -28,6 +28,14 @@
|
|
| 28 |
720,
|
| 29 |
1280
|
| 30 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
}
|
| 32 |
},
|
| 33 |
"output_features": {
|
|
@@ -41,7 +49,7 @@
|
|
| 41 |
"device": "cuda",
|
| 42 |
"use_amp": false,
|
| 43 |
"push_to_hub": true,
|
| 44 |
-
"repo_id": "LeoMuuuu/hanoi-
|
| 45 |
"private": null,
|
| 46 |
"tags": null,
|
| 47 |
"license": null,
|
|
|
|
| 28 |
720,
|
| 29 |
1280
|
| 30 |
]
|
| 31 |
+
},
|
| 32 |
+
"observation.images.front": {
|
| 33 |
+
"type": "VISUAL",
|
| 34 |
+
"shape": [
|
| 35 |
+
3,
|
| 36 |
+
720,
|
| 37 |
+
1280
|
| 38 |
+
]
|
| 39 |
}
|
| 40 |
},
|
| 41 |
"output_features": {
|
|
|
|
| 49 |
"device": "cuda",
|
| 50 |
"use_amp": false,
|
| 51 |
"push_to_hub": true,
|
| 52 |
+
"repo_id": "LeoMuuuu/hanoi-red2middle",
|
| 53 |
"private": null,
|
| 54 |
"tags": null,
|
| 55 |
"license": null,
|
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:3c88a7090891deae6780f0ccb9aa2b48e59e00848ec156f1ab9e1618fc09e504
|
| 3 |
+
size 206701344
|
train_config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
-
"repo_id": "LeoMuuuu/hanoi-
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
|
@@ -95,6 +95,14 @@
|
|
| 95 |
720,
|
| 96 |
1280
|
| 97 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 98 |
}
|
| 99 |
},
|
| 100 |
"output_features": {
|
|
@@ -108,7 +116,7 @@
|
|
| 108 |
"device": "cuda",
|
| 109 |
"use_amp": false,
|
| 110 |
"push_to_hub": true,
|
| 111 |
-
"repo_id": "LeoMuuuu/hanoi-
|
| 112 |
"private": null,
|
| 113 |
"tags": null,
|
| 114 |
"license": null,
|
|
@@ -134,12 +142,12 @@
|
|
| 134 |
"optimizer_weight_decay": 0.0001,
|
| 135 |
"optimizer_lr_backbone": 1e-05
|
| 136 |
},
|
| 137 |
-
"output_dir": "outputs/train/hanoi-
|
| 138 |
-
"job_name": "hanoi-
|
| 139 |
"resume": false,
|
| 140 |
"seed": 1000,
|
| 141 |
"num_workers": 4,
|
| 142 |
-
"batch_size":
|
| 143 |
"steps": 100000,
|
| 144 |
"eval_freq": 20000,
|
| 145 |
"log_freq": 200,
|
|
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
+
"repo_id": "LeoMuuuu/hanoi-red2middle",
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
|
|
|
| 95 |
720,
|
| 96 |
1280
|
| 97 |
]
|
| 98 |
+
},
|
| 99 |
+
"observation.images.front": {
|
| 100 |
+
"type": "VISUAL",
|
| 101 |
+
"shape": [
|
| 102 |
+
3,
|
| 103 |
+
720,
|
| 104 |
+
1280
|
| 105 |
+
]
|
| 106 |
}
|
| 107 |
},
|
| 108 |
"output_features": {
|
|
|
|
| 116 |
"device": "cuda",
|
| 117 |
"use_amp": false,
|
| 118 |
"push_to_hub": true,
|
| 119 |
+
"repo_id": "LeoMuuuu/hanoi-red2middle",
|
| 120 |
"private": null,
|
| 121 |
"tags": null,
|
| 122 |
"license": null,
|
|
|
|
| 142 |
"optimizer_weight_decay": 0.0001,
|
| 143 |
"optimizer_lr_backbone": 1e-05
|
| 144 |
},
|
| 145 |
+
"output_dir": "outputs/train/hanoi-red2middle",
|
| 146 |
+
"job_name": "hanoi-red2middle",
|
| 147 |
"resume": false,
|
| 148 |
"seed": 1000,
|
| 149 |
"num_workers": 4,
|
| 150 |
+
"batch_size": 4,
|
| 151 |
"steps": 100000,
|
| 152 |
"eval_freq": 20000,
|
| 153 |
"log_freq": 200,
|