Instructions to use open-cloth/diffusion-cloth with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- LeRobot
How to use open-cloth/diffusion-cloth with LeRobot:
- Notebooks
- Google Colab
- Kaggle
Upload policy weights, train config and readme
Browse files- README.md +1 -1
- model.safetensors +1 -1
- train_config.json +7 -5
README.md
CHANGED
|
@@ -5,9 +5,9 @@ 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
|
|
|
|
| 5 |
model_name: diffusion
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
+
- robotics
|
| 9 |
- diffusion
|
| 10 |
- lerobot
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for diffusion
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 312295392
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1cb65c1c8a262a8711fa57277fe7fa363fb5d6d835f6e750d5d404b5ced403ec
|
| 3 |
size 312295392
|
train_config.json
CHANGED
|
@@ -170,8 +170,8 @@
|
|
| 170 |
"scheduler_name": "cosine",
|
| 171 |
"scheduler_warmup_steps": 500
|
| 172 |
},
|
| 173 |
-
"output_dir": "outputs/train/
|
| 174 |
-
"job_name": "
|
| 175 |
"resume": false,
|
| 176 |
"seed": 1000,
|
| 177 |
"cudnn_deterministic": false,
|
|
@@ -208,10 +208,10 @@
|
|
| 208 |
"wandb": {
|
| 209 |
"enable": true,
|
| 210 |
"disable_artifact": false,
|
| 211 |
-
"project": "
|
| 212 |
"entity": null,
|
| 213 |
"notes": null,
|
| 214 |
-
"run_id": "
|
| 215 |
"mode": null,
|
| 216 |
"add_tags": true
|
| 217 |
},
|
|
@@ -221,6 +221,8 @@
|
|
| 221 |
"rabc_kappa": 0.01,
|
| 222 |
"rabc_epsilon": 1e-06,
|
| 223 |
"rabc_head_mode": "sparse",
|
| 224 |
-
"rename_map": {
|
|
|
|
|
|
|
| 225 |
"checkpoint_path": null
|
| 226 |
}
|
|
|
|
| 170 |
"scheduler_name": "cosine",
|
| 171 |
"scheduler_warmup_steps": 500
|
| 172 |
},
|
| 173 |
+
"output_dir": "outputs/train/diffusion",
|
| 174 |
+
"job_name": "diffusion",
|
| 175 |
"resume": false,
|
| 176 |
"seed": 1000,
|
| 177 |
"cudnn_deterministic": false,
|
|
|
|
| 208 |
"wandb": {
|
| 209 |
"enable": true,
|
| 210 |
"disable_artifact": false,
|
| 211 |
+
"project": "open-cloth",
|
| 212 |
"entity": null,
|
| 213 |
"notes": null,
|
| 214 |
+
"run_id": "j8xy94cl",
|
| 215 |
"mode": null,
|
| 216 |
"add_tags": true
|
| 217 |
},
|
|
|
|
| 221 |
"rabc_kappa": 0.01,
|
| 222 |
"rabc_epsilon": 1e-06,
|
| 223 |
"rabc_head_mode": "sparse",
|
| 224 |
+
"rename_map": {
|
| 225 |
+
"observation.images.gripper": "observation.images.camera1"
|
| 226 |
+
},
|
| 227 |
"checkpoint_path": null
|
| 228 |
}
|