Instructions to use giacomoran/so101_gba_dm_act_smooth_p4f2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- LeRobot
How to use giacomoran/so101_gba_dm_act_smooth_p4f2 with LeRobot:
- Notebooks
- Google Colab
- Kaggle
Upload policy weights, train config and readme
Browse files- config.json +1 -1
- model.safetensors +1 -1
- train_config.json +4 -4
config.json
CHANGED
|
@@ -27,7 +27,7 @@
|
|
| 27 |
"private": null,
|
| 28 |
"tags": null,
|
| 29 |
"license": null,
|
| 30 |
-
"pretrained_path": "/workspace/so101_gba_dm_act_smooth_p4f2/checkpoints/
|
| 31 |
"chunk_size": 30,
|
| 32 |
"n_action_steps": 30,
|
| 33 |
"length_prefix_past": 4,
|
|
|
|
| 27 |
"private": null,
|
| 28 |
"tags": null,
|
| 29 |
"license": null,
|
| 30 |
+
"pretrained_path": "/workspace/so101_gba_dm_act_smooth_p4f2/checkpoints/030003/pretrained_model",
|
| 31 |
"chunk_size": 30,
|
| 32 |
"n_action_steps": 30,
|
| 33 |
"length_prefix_past": 4,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 241960176
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a7b993c40c4ecd5076acffa369798740a9dafcad9c347cc12a974bfb6223187c
|
| 3 |
size 241960176
|
train_config.json
CHANGED
|
@@ -109,7 +109,7 @@
|
|
| 109 |
"private": null,
|
| 110 |
"tags": null,
|
| 111 |
"license": null,
|
| 112 |
-
"pretrained_path": "/workspace/so101_gba_dm_act_smooth_p4f2/checkpoints/
|
| 113 |
"chunk_size": 30,
|
| 114 |
"n_action_steps": 30,
|
| 115 |
"length_prefix_past": 4,
|
|
@@ -146,12 +146,12 @@
|
|
| 146 |
"cudnn_deterministic": false,
|
| 147 |
"num_workers": 8,
|
| 148 |
"batch_size": 32,
|
| 149 |
-
"steps":
|
| 150 |
"eval_freq": 20000,
|
| 151 |
"log_freq": 200,
|
| 152 |
"tolerance_s": 0.0001,
|
| 153 |
"save_checkpoint": true,
|
| 154 |
-
"save_freq":
|
| 155 |
"use_policy_training_preset": true,
|
| 156 |
"optimizer": {
|
| 157 |
"type": "adamw",
|
|
@@ -186,5 +186,5 @@
|
|
| 186 |
"rabc_epsilon": 1e-06,
|
| 187 |
"rabc_head_mode": "sparse",
|
| 188 |
"rename_map": {},
|
| 189 |
-
"checkpoint_path": "/workspace/so101_gba_dm_act_smooth_p4f2/checkpoints/
|
| 190 |
}
|
|
|
|
| 109 |
"private": null,
|
| 110 |
"tags": null,
|
| 111 |
"license": null,
|
| 112 |
+
"pretrained_path": "/workspace/so101_gba_dm_act_smooth_p4f2/checkpoints/030003/pretrained_model",
|
| 113 |
"chunk_size": 30,
|
| 114 |
"n_action_steps": 30,
|
| 115 |
"length_prefix_past": 4,
|
|
|
|
| 146 |
"cudnn_deterministic": false,
|
| 147 |
"num_workers": 8,
|
| 148 |
"batch_size": 32,
|
| 149 |
+
"steps": 100000,
|
| 150 |
"eval_freq": 20000,
|
| 151 |
"log_freq": 200,
|
| 152 |
"tolerance_s": 0.0001,
|
| 153 |
"save_checkpoint": true,
|
| 154 |
+
"save_freq": 10000,
|
| 155 |
"use_policy_training_preset": true,
|
| 156 |
"optimizer": {
|
| 157 |
"type": "adamw",
|
|
|
|
| 186 |
"rabc_epsilon": 1e-06,
|
| 187 |
"rabc_head_mode": "sparse",
|
| 188 |
"rename_map": {},
|
| 189 |
+
"checkpoint_path": "/workspace/so101_gba_dm_act_smooth_p4f2/checkpoints/030003"
|
| 190 |
}
|