act_ring_insert / README.md
pravsels's picture
Update model card
e037b0c verified
|
Raw
History Blame Contribute Delete
1.12 kB
---
library_name: lerobot
tags:
- act
- robotics
- lerobot
---
# act_ring_insert
Standard LeRobot ACT trained from scratch on SO101 data.
| | |
|---|---|
| **Policy** | ACT (`policy.type=act`) |
| **Dataset** | [villekuosmanen/armnetbench_ring_insert](https://huggingface.co/datasets/villekuosmanen/armnetbench_ring_insert) |
| **Task** | `ring_insert` |
| **Action dim** | 6 (single-arm) |
| **Cameras** | `top`, `wrist`, `front` |
| **Training** | GCloud A100-80GB (alpha-act-a100-1), batch 16, action horizon 30 |
| **Loss @ 10k** | 0.163 |
| **W&B project** | [act_ring_insert](https://wandb.ai/pravsels/act_ring_insert) |
| **W&B run** | [swmdjqdn](https://wandb.ai/pravsels/act_ring_insert/runs/swmdjqdn) |
## Checkpoints
Archived copies:
| Step | Path |
|------|------|
| 010000 | `checkpoints/010000/pretrained_model/` |
| 008000 | `checkpoints/008000/pretrained_model/` |
The latest step (`010000`) is also copied to the **repository root** for direct loading.
## Usage
```python
from lerobot.policies.act.modeling_act import ACTPolicy
policy = ACTPolicy.from_pretrained("pravsels/act_ring_insert")
```