Upload policy weights, train config and readme
Browse files- README.md +1 -1
- config.json +1 -1
- model.safetensors +1 -1
- train_config.json +2 -2
README.md
CHANGED
|
@@ -6,8 +6,8 @@ model_name: diffusion
|
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
- lerobot
|
| 9 |
-
- robotics
|
| 10 |
- diffusion
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for diffusion
|
|
|
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
- lerobot
|
|
|
|
| 9 |
- diffusion
|
| 10 |
+
- robotics
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for diffusion
|
config.json
CHANGED
|
@@ -38,7 +38,7 @@
|
|
| 38 |
]
|
| 39 |
}
|
| 40 |
},
|
| 41 |
-
"device": "cuda:
|
| 42 |
"use_amp": false,
|
| 43 |
"push_to_hub": true,
|
| 44 |
"repo_id": "y1y2y3/so101_test4_diffusion_12k",
|
|
|
|
| 38 |
]
|
| 39 |
}
|
| 40 |
},
|
| 41 |
+
"device": "cuda:0",
|
| 42 |
"use_amp": false,
|
| 43 |
"push_to_hub": true,
|
| 44 |
"repo_id": "y1y2y3/so101_test4_diffusion_12k",
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1066517744
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b195a81ed7a3be551c2290ca967580f8dadb52b78cd0f92ad960115cccf7c624
|
| 3 |
size 1066517744
|
train_config.json
CHANGED
|
@@ -105,7 +105,7 @@
|
|
| 105 |
]
|
| 106 |
}
|
| 107 |
},
|
| 108 |
-
"device": "cuda:
|
| 109 |
"use_amp": false,
|
| 110 |
"push_to_hub": true,
|
| 111 |
"repo_id": "y1y2y3/so101_test4_diffusion_12k",
|
|
@@ -193,7 +193,7 @@
|
|
| 193 |
"project": "lerobot",
|
| 194 |
"entity": null,
|
| 195 |
"notes": null,
|
| 196 |
-
"run_id":
|
| 197 |
"mode": null
|
| 198 |
}
|
| 199 |
}
|
|
|
|
| 105 |
]
|
| 106 |
}
|
| 107 |
},
|
| 108 |
+
"device": "cuda:0",
|
| 109 |
"use_amp": false,
|
| 110 |
"push_to_hub": true,
|
| 111 |
"repo_id": "y1y2y3/so101_test4_diffusion_12k",
|
|
|
|
| 193 |
"project": "lerobot",
|
| 194 |
"entity": null,
|
| 195 |
"notes": null,
|
| 196 |
+
"run_id": "d6glqpex",
|
| 197 |
"mode": null
|
| 198 |
}
|
| 199 |
}
|