Upload policy weights, train config and readme
Browse files- README.md +3 -3
- config.json +1 -0
- model.safetensors +2 -2
- train_config.json +5 -3
README.md
CHANGED
|
@@ -6,8 +6,8 @@ model_name: act
|
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
- robotics
|
| 9 |
-
- act
|
| 10 |
- lerobot
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for act
|
|
@@ -31,7 +31,7 @@ Below is the short version on how to train and run inference/eval:
|
|
| 31 |
### Train from scratch
|
| 32 |
|
| 33 |
```bash
|
| 34 |
-
|
| 35 |
--dataset.repo_id=${HF_USER}/<dataset> \
|
| 36 |
--policy.type=act \
|
| 37 |
--output_dir=outputs/train/<desired_policy_repo_id> \
|
|
@@ -46,7 +46,7 @@ _Writes checkpoints to `outputs/train/<desired_policy_repo_id>/checkpoints/`._
|
|
| 46 |
### Evaluate the policy/run inference
|
| 47 |
|
| 48 |
```bash
|
| 49 |
-
|
| 50 |
--robot.type=so100_follower \
|
| 51 |
--dataset.repo_id=<hf_user>/eval_<dataset> \
|
| 52 |
--policy.path=<hf_user>/<desired_policy_repo_id> \
|
|
|
|
| 6 |
pipeline_tag: robotics
|
| 7 |
tags:
|
| 8 |
- robotics
|
|
|
|
| 9 |
- lerobot
|
| 10 |
+
- act
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for act
|
|
|
|
| 31 |
### Train from scratch
|
| 32 |
|
| 33 |
```bash
|
| 34 |
+
lerobot-train \
|
| 35 |
--dataset.repo_id=${HF_USER}/<dataset> \
|
| 36 |
--policy.type=act \
|
| 37 |
--output_dir=outputs/train/<desired_policy_repo_id> \
|
|
|
|
| 46 |
### Evaluate the policy/run inference
|
| 47 |
|
| 48 |
```bash
|
| 49 |
+
lerobot-record \
|
| 50 |
--robot.type=so100_follower \
|
| 51 |
--dataset.repo_id=<hf_user>/eval_<dataset> \
|
| 52 |
--policy.path=<hf_user>/<desired_policy_repo_id> \
|
config.json
CHANGED
|
@@ -40,6 +40,7 @@
|
|
| 40 |
},
|
| 41 |
"device": "cuda",
|
| 42 |
"use_amp": false,
|
|
|
|
| 43 |
"push_to_hub": true,
|
| 44 |
"repo_id": "fitrijamat/pick-insert-block",
|
| 45 |
"private": null,
|
|
|
|
| 40 |
},
|
| 41 |
"device": "cuda",
|
| 42 |
"use_amp": false,
|
| 43 |
+
"gradient_accumulation_steps": 1,
|
| 44 |
"push_to_hub": true,
|
| 45 |
"repo_id": "fitrijamat/pick-insert-block",
|
| 46 |
"private": null,
|
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:00c3c9df6ba6ddaa7fa2623da41cc0b27639c401f0be7c2dad2d994f4380288d
|
| 3 |
+
size 206699736
|
train_config.json
CHANGED
|
@@ -62,7 +62,8 @@
|
|
| 62 |
},
|
| 63 |
"revision": "main",
|
| 64 |
"use_imagenet_stats": true,
|
| 65 |
-
"video_backend": "pyav"
|
|
|
|
| 66 |
},
|
| 67 |
"env": null,
|
| 68 |
"policy": {
|
|
@@ -107,6 +108,7 @@
|
|
| 107 |
},
|
| 108 |
"device": "cuda",
|
| 109 |
"use_amp": false,
|
|
|
|
| 110 |
"push_to_hub": true,
|
| 111 |
"repo_id": "fitrijamat/pick-insert-block",
|
| 112 |
"private": null,
|
|
@@ -134,7 +136,7 @@
|
|
| 134 |
"optimizer_weight_decay": 0.0001,
|
| 135 |
"optimizer_lr_backbone": 1e-05
|
| 136 |
},
|
| 137 |
-
"output_dir": "C:\\Users\\User\\Documents\\reuben_ws\\lerobot\\outputs\\train\\
|
| 138 |
"job_name": "act_pick-insert-block",
|
| 139 |
"resume": false,
|
| 140 |
"seed": 1000,
|
|
@@ -169,7 +171,7 @@
|
|
| 169 |
"project": "lerobot",
|
| 170 |
"entity": null,
|
| 171 |
"notes": null,
|
| 172 |
-
"run_id": "
|
| 173 |
"mode": null
|
| 174 |
}
|
| 175 |
}
|
|
|
|
| 62 |
},
|
| 63 |
"revision": "main",
|
| 64 |
"use_imagenet_stats": true,
|
| 65 |
+
"video_backend": "pyav",
|
| 66 |
+
"streaming": false
|
| 67 |
},
|
| 68 |
"env": null,
|
| 69 |
"policy": {
|
|
|
|
| 108 |
},
|
| 109 |
"device": "cuda",
|
| 110 |
"use_amp": false,
|
| 111 |
+
"gradient_accumulation_steps": 1,
|
| 112 |
"push_to_hub": true,
|
| 113 |
"repo_id": "fitrijamat/pick-insert-block",
|
| 114 |
"private": null,
|
|
|
|
| 136 |
"optimizer_weight_decay": 0.0001,
|
| 137 |
"optimizer_lr_backbone": 1e-05
|
| 138 |
},
|
| 139 |
+
"output_dir": "C:\\Users\\User\\Documents\\reuben_ws\\lerobot\\outputs\\train\\act_mini_studio_block_in_cup_latest",
|
| 140 |
"job_name": "act_pick-insert-block",
|
| 141 |
"resume": false,
|
| 142 |
"seed": 1000,
|
|
|
|
| 171 |
"project": "lerobot",
|
| 172 |
"entity": null,
|
| 173 |
"notes": null,
|
| 174 |
+
"run_id": "edc35ocv",
|
| 175 |
"mode": null
|
| 176 |
}
|
| 177 |
}
|