Upload policy weights, train config and readme
Browse files- README.md +4 -4
- config.json +5 -2
- model.safetensors +2 -2
- train_config.json +9 -6
README.md
CHANGED
|
@@ -2,15 +2,15 @@
|
|
| 2 |
datasets: quincyu/trace_robotwin
|
| 3 |
library_name: lerobot
|
| 4 |
license: apache-2.0
|
| 5 |
-
model_name:
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
-
-
|
| 9 |
-
- lerobot
|
| 10 |
- robotics
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
-
# Model Card for
|
| 14 |
|
| 15 |
<!-- Provide a quick summary of what the model is/does. -->
|
| 16 |
|
|
|
|
| 2 |
datasets: quincyu/trace_robotwin
|
| 3 |
library_name: lerobot
|
| 4 |
license: apache-2.0
|
| 5 |
+
model_name: tg_act_266m
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
+
- tg_act_266m
|
|
|
|
| 9 |
- robotics
|
| 10 |
+
- lerobot
|
| 11 |
---
|
| 12 |
|
| 13 |
+
# Model Card for tg_act_266m
|
| 14 |
|
| 15 |
<!-- Provide a quick summary of what the model is/does. -->
|
| 16 |
|
config.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"type": "
|
| 3 |
"n_obs_steps": 1,
|
| 4 |
"input_features": {
|
| 5 |
"observation.state": {
|
|
@@ -64,6 +64,7 @@
|
|
| 64 |
"pretrained_path": null,
|
| 65 |
"chunk_size": 100,
|
| 66 |
"n_action_steps": 100,
|
|
|
|
| 67 |
"normalization_mapping": {
|
| 68 |
"VISUAL": "MEAN_STD",
|
| 69 |
"STATE": "MEAN_STD",
|
|
@@ -87,5 +88,7 @@
|
|
| 87 |
"kl_weight": 10.0,
|
| 88 |
"optimizer_lr": 1e-05,
|
| 89 |
"optimizer_weight_decay": 0.0001,
|
| 90 |
-
"optimizer_lr_backbone": 1e-05
|
|
|
|
|
|
|
| 91 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"type": "tg_act_266m",
|
| 3 |
"n_obs_steps": 1,
|
| 4 |
"input_features": {
|
| 5 |
"observation.state": {
|
|
|
|
| 64 |
"pretrained_path": null,
|
| 65 |
"chunk_size": 100,
|
| 66 |
"n_action_steps": 100,
|
| 67 |
+
"ee_pixel_image_resolution": 256,
|
| 68 |
"normalization_mapping": {
|
| 69 |
"VISUAL": "MEAN_STD",
|
| 70 |
"STATE": "MEAN_STD",
|
|
|
|
| 88 |
"kl_weight": 10.0,
|
| 89 |
"optimizer_lr": 1e-05,
|
| 90 |
"optimizer_weight_decay": 0.0001,
|
| 91 |
+
"optimizer_lr_backbone": 1e-05,
|
| 92 |
+
"compile_model": false,
|
| 93 |
+
"compile_mode": "max-autotune"
|
| 94 |
}
|
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:55436b7f64d5a7bd442085344e0e8aa6096120fedd37b830601a3186a39dee56
|
| 3 |
+
size 1064876344
|
train_config.json
CHANGED
|
@@ -81,7 +81,7 @@
|
|
| 81 |
},
|
| 82 |
"env": null,
|
| 83 |
"policy": {
|
| 84 |
-
"type": "
|
| 85 |
"n_obs_steps": 1,
|
| 86 |
"input_features": {
|
| 87 |
"observation.state": {
|
|
@@ -146,6 +146,7 @@
|
|
| 146 |
"pretrained_path": null,
|
| 147 |
"chunk_size": 100,
|
| 148 |
"n_action_steps": 100,
|
|
|
|
| 149 |
"normalization_mapping": {
|
| 150 |
"VISUAL": "MEAN_STD",
|
| 151 |
"STATE": "MEAN_STD",
|
|
@@ -169,13 +170,15 @@
|
|
| 169 |
"kl_weight": 10.0,
|
| 170 |
"optimizer_lr": 1e-05,
|
| 171 |
"optimizer_weight_decay": 0.0001,
|
| 172 |
-
"optimizer_lr_backbone": 1e-05
|
|
|
|
|
|
|
| 173 |
},
|
| 174 |
-
"output_dir": "outputs/train/2026-02-
|
| 175 |
-
"job_name": "
|
| 176 |
"resume": false,
|
| 177 |
"seed": 0,
|
| 178 |
-
"num_workers":
|
| 179 |
"batch_size": 128,
|
| 180 |
"steps": 100000,
|
| 181 |
"eval_freq": 20000,
|
|
@@ -207,7 +210,7 @@
|
|
| 207 |
"project": "lerobot",
|
| 208 |
"entity": null,
|
| 209 |
"notes": null,
|
| 210 |
-
"run_id": "
|
| 211 |
"mode": null
|
| 212 |
},
|
| 213 |
"peft": null,
|
|
|
|
| 81 |
},
|
| 82 |
"env": null,
|
| 83 |
"policy": {
|
| 84 |
+
"type": "tg_act_266m",
|
| 85 |
"n_obs_steps": 1,
|
| 86 |
"input_features": {
|
| 87 |
"observation.state": {
|
|
|
|
| 146 |
"pretrained_path": null,
|
| 147 |
"chunk_size": 100,
|
| 148 |
"n_action_steps": 100,
|
| 149 |
+
"ee_pixel_image_resolution": 256,
|
| 150 |
"normalization_mapping": {
|
| 151 |
"VISUAL": "MEAN_STD",
|
| 152 |
"STATE": "MEAN_STD",
|
|
|
|
| 170 |
"kl_weight": 10.0,
|
| 171 |
"optimizer_lr": 1e-05,
|
| 172 |
"optimizer_weight_decay": 0.0001,
|
| 173 |
+
"optimizer_lr_backbone": 1e-05,
|
| 174 |
+
"compile_model": false,
|
| 175 |
+
"compile_mode": "max-autotune"
|
| 176 |
},
|
| 177 |
+
"output_dir": "outputs/train/2026-02-12/19-23-15_tg_act_266m",
|
| 178 |
+
"job_name": "tg_act_266m",
|
| 179 |
"resume": false,
|
| 180 |
"seed": 0,
|
| 181 |
+
"num_workers": 4,
|
| 182 |
"batch_size": 128,
|
| 183 |
"steps": 100000,
|
| 184 |
"eval_freq": 20000,
|
|
|
|
| 210 |
"project": "lerobot",
|
| 211 |
"entity": null,
|
| 212 |
"notes": null,
|
| 213 |
+
"run_id": "r5hwbuha",
|
| 214 |
"mode": null
|
| 215 |
},
|
| 216 |
"peft": null,
|