Upload policy weights, train config and readme
Browse files- README.md +3 -3
- config.json +2 -2
- model.safetensors +1 -1
- train_config.json +8 -8
README.md
CHANGED
|
@@ -1,13 +1,13 @@
|
|
| 1 |
---
|
| 2 |
-
datasets:
|
| 3 |
library_name: lerobot
|
| 4 |
license: apache-2.0
|
| 5 |
model_name: diffusion
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
-
- robotics
|
| 9 |
-
- lerobot
|
| 10 |
- diffusion
|
|
|
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for diffusion
|
|
|
|
| 1 |
---
|
| 2 |
+
datasets: pusht_merged_300_fine
|
| 3 |
library_name: lerobot
|
| 4 |
license: apache-2.0
|
| 5 |
model_name: diffusion
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
|
|
|
|
|
|
| 8 |
- diffusion
|
| 9 |
+
- lerobot
|
| 10 |
+
- robotics
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for diffusion
|
config.json
CHANGED
|
@@ -36,7 +36,7 @@
|
|
| 36 |
"normalization_mapping": {
|
| 37 |
"VISUAL": "MEAN_STD",
|
| 38 |
"STATE": "MIN_MAX",
|
| 39 |
-
"ACTION": "
|
| 40 |
"ENV": "MIN_MAX"
|
| 41 |
},
|
| 42 |
"drop_n_last_frames": 7,
|
|
@@ -70,7 +70,7 @@
|
|
| 70 |
"clip_sample_range": 1.0,
|
| 71 |
"num_inference_steps": null,
|
| 72 |
"do_mask_loss_for_padding": false,
|
| 73 |
-
"optimizer_lr":
|
| 74 |
"optimizer_betas": [
|
| 75 |
0.95,
|
| 76 |
0.999
|
|
|
|
| 36 |
"normalization_mapping": {
|
| 37 |
"VISUAL": "MEAN_STD",
|
| 38 |
"STATE": "MIN_MAX",
|
| 39 |
+
"ACTION": "MIN_MAX",
|
| 40 |
"ENV": "MIN_MAX"
|
| 41 |
},
|
| 42 |
"drop_n_last_frames": 7,
|
|
|
|
| 70 |
"clip_sample_range": 1.0,
|
| 71 |
"num_inference_steps": null,
|
| 72 |
"do_mask_loss_for_padding": false,
|
| 73 |
+
"optimizer_lr": 0.0002,
|
| 74 |
"optimizer_betas": [
|
| 75 |
0.95,
|
| 76 |
0.999
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 995055496
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9ee1307bf7930b9ffb9eac5c9720025a58df265163e09a95f6d9f8f000955c99
|
| 3 |
size 995055496
|
train_config.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
-
"repo_id": "
|
| 4 |
-
"root": "pusht_teleop_data/
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
| 7 |
"enable": false,
|
|
@@ -132,7 +132,7 @@
|
|
| 132 |
"gaussian_blur": null,
|
| 133 |
"CoverGreenT": null,
|
| 134 |
"coarsity": "fine",
|
| 135 |
-
"action_type": "
|
| 136 |
},
|
| 137 |
"policy": {
|
| 138 |
"type": "diffusion",
|
|
@@ -172,7 +172,7 @@
|
|
| 172 |
"normalization_mapping": {
|
| 173 |
"VISUAL": "MEAN_STD",
|
| 174 |
"STATE": "MIN_MAX",
|
| 175 |
-
"ACTION": "
|
| 176 |
"ENV": "MIN_MAX"
|
| 177 |
},
|
| 178 |
"drop_n_last_frames": 7,
|
|
@@ -206,7 +206,7 @@
|
|
| 206 |
"clip_sample_range": 1.0,
|
| 207 |
"num_inference_steps": null,
|
| 208 |
"do_mask_loss_for_padding": false,
|
| 209 |
-
"optimizer_lr":
|
| 210 |
"optimizer_betas": [
|
| 211 |
0.95,
|
| 212 |
0.999
|
|
@@ -218,7 +218,7 @@
|
|
| 218 |
"partial_green_t_cover_processor": false,
|
| 219 |
"use_rnd": false
|
| 220 |
},
|
| 221 |
-
"output_dir": "outputs/train/2025-12-09
|
| 222 |
"job_name": "pusht_diffusion_fine",
|
| 223 |
"resume": false,
|
| 224 |
"seed": 1000,
|
|
@@ -232,7 +232,7 @@
|
|
| 232 |
"use_policy_training_preset": true,
|
| 233 |
"optimizer": {
|
| 234 |
"type": "adam",
|
| 235 |
-
"lr":
|
| 236 |
"weight_decay": 1e-06,
|
| 237 |
"grad_clip_norm": 10.0,
|
| 238 |
"betas": [
|
|
@@ -259,7 +259,7 @@
|
|
| 259 |
"project": "policies",
|
| 260 |
"entity": null,
|
| 261 |
"notes": null,
|
| 262 |
-
"run_id": "
|
| 263 |
"mode": null
|
| 264 |
},
|
| 265 |
"checkpoint_path": null,
|
|
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
+
"repo_id": "pusht_merged_300_fine",
|
| 4 |
+
"root": "pusht_teleop_data/pusht_merged_300_fine",
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
| 7 |
"enable": false,
|
|
|
|
| 132 |
"gaussian_blur": null,
|
| 133 |
"CoverGreenT": null,
|
| 134 |
"coarsity": "fine",
|
| 135 |
+
"action_type": "absolute"
|
| 136 |
},
|
| 137 |
"policy": {
|
| 138 |
"type": "diffusion",
|
|
|
|
| 172 |
"normalization_mapping": {
|
| 173 |
"VISUAL": "MEAN_STD",
|
| 174 |
"STATE": "MIN_MAX",
|
| 175 |
+
"ACTION": "MIN_MAX",
|
| 176 |
"ENV": "MIN_MAX"
|
| 177 |
},
|
| 178 |
"drop_n_last_frames": 7,
|
|
|
|
| 206 |
"clip_sample_range": 1.0,
|
| 207 |
"num_inference_steps": null,
|
| 208 |
"do_mask_loss_for_padding": false,
|
| 209 |
+
"optimizer_lr": 0.0002,
|
| 210 |
"optimizer_betas": [
|
| 211 |
0.95,
|
| 212 |
0.999
|
|
|
|
| 218 |
"partial_green_t_cover_processor": false,
|
| 219 |
"use_rnd": false
|
| 220 |
},
|
| 221 |
+
"output_dir": "outputs/train/2025-12-10/09-43-55_pusht_diffusion_fine",
|
| 222 |
"job_name": "pusht_diffusion_fine",
|
| 223 |
"resume": false,
|
| 224 |
"seed": 1000,
|
|
|
|
| 232 |
"use_policy_training_preset": true,
|
| 233 |
"optimizer": {
|
| 234 |
"type": "adam",
|
| 235 |
+
"lr": 0.0002,
|
| 236 |
"weight_decay": 1e-06,
|
| 237 |
"grad_clip_norm": 10.0,
|
| 238 |
"betas": [
|
|
|
|
| 259 |
"project": "policies",
|
| 260 |
"entity": null,
|
| 261 |
"notes": null,
|
| 262 |
+
"run_id": "aux7jvgy",
|
| 263 |
"mode": null
|
| 264 |
},
|
| 265 |
"checkpoint_path": null,
|