Hugging Face's logo

AdamLucek
/
act_koch_block

Robotics
Safetensors
lerobot
act
Model card Files Community
act_koch_block / README.md
AdamLucek's picture
AdamLucek
Update README.md
f1af368 verified
preview code
|
raw
history blame contribute delete
1.39 kB
metadata
library_name: lerobot
tags:
  - act
  - robotics
  - lerobot
license: mit
datasets:
  - AdamLucek/koch_block

Action Chunking Transformer For The Koch v1.1 Robot

Action Chunking Transformer (ACT) trained on the AdamLucek/koch_block dataset for the Koch v1.1 Low Cost Robot from LeRobot

Usage

Details around methodology for set up, training, and evaluation available via LeRobot's guide to Getting Started with Real World Robots

Training Details

The model was trained using LeRobot's training script and the AdamLucek/koch_block dataset, using this command:

python lerobot/scripts/train.py `
  dataset_repo_id="AdamLucek/koch_block" `
  policy="act_koch_real" `
  env="koch_real" `
  hydra.run.dir="outputs/train/act_koch_test" `
  hydra.job.name="act_koch_test" `
  device="cuda" `
  wandb.enable="true"

Training was done on a 3060 for roughly 7 hours, loss curve and statistics available in wandb report here.