Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add koch bimanual #385

Merged
merged 3 commits into from
Aug 27, 2024
Merged

Conversation

Cadene
Copy link
Collaborator

@Cadene Cadene commented Aug 27, 2024

What this does

Title

How it was tested

Folded some shirts: https://x.com/RemiCadene/status/1828006677528137977

Visualized the dataset
https://huggingface.co/spaces/cadene/visualize_dataset_koch_bimanual_folding

How to checkout & try? (for the reviewer)

python lerobot/scripts/control_robot.py teleoperate \
  --robot-path lerobot/configs/robot/koch_bimanual.yaml 

python lerobot/scripts/train.py \
  dataset_repo_id=cadene/koch_bimanual_folding \
  policy=act_koch_real \
  env=koch_real \
  hydra.run.dir=outputs/train/act_koch_bimanual_folding \
  hydra.job.name=act_koch_bimanual_folding \
  device=cuda \
  wandb.enable=true \
  env.state_dim=12 \
  env.action_dim=12 \
  training.num_workers=8 \
  training.batch_size=32 \
  policy.n_decoder_layers=4

huggingface-cli upload cadene/act_koch_bimanual_folding \
  outputs/train/act_koch_bimanual_folding/checkpoints/last/pretrained_model

python lerobot/scripts/control_robot.py record \
  --robot-path lerobot/configs/robot/koch_bimanual.yaml \
  --fps 30 \
  --root data \
  --repo-id cadene/eval_act_koch_bimanual_folding_080000 \
  --num-episodes 4 \
  --run-compute-stats 0 \
  --warmup-time-s 10 \
  --episode-time-s 360 \
  --reset-time-s 360 \
  --force-override 0 \
  -p cadene/act_koch_bimanual_folding_080000 \
  --policy-overrides device=mps

python lerobot/scripts/visualize_dataset_html.py \
  --root data \
  --repo-id cadene/eval_act_koch_bimanual_folding_080000

@Cadene Cadene requested a review from alexander-soare August 27, 2024 20:03
@Cadene Cadene force-pushed the user/rcadene/2024_08_25_koch_bimanual_folding branch from 4e964be to 09dc752 Compare August 27, 2024 20:03
@Cadene Cadene marked this pull request as ready for review August 27, 2024 20:03
@Cadene Cadene merged commit 1ce418e into main Aug 27, 2024
7 checks passed
@Cadene Cadene deleted the user/rcadene/2024_08_25_koch_bimanual_folding branch August 27, 2024 22:53
amandip7 pushed a commit to amandip7/lerobot that referenced this pull request Oct 10, 2024
menhguin pushed a commit to menhguin/lerobot that referenced this pull request Feb 9, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants