makermods/arrange_chocolate_finetune
Viewer • Updated • 62.4k • 349
How to use makermods/smolvla_finetune_arrange_chocolate with LeRobot:
# See https://github.com/huggingface/lerobot?tab=readme-ov-file#installation for more details git clone https://github.com/huggingface/lerobot.git cd lerobot pip install -e .[smolvla]
# Launch finetuning on your dataset python lerobot/scripts/train.py \ --policy.path=makermods/smolvla_finetune_arrange_chocolate \ --dataset.repo_id=lerobot/svla_so101_pickplace \ --batch_size=64 \ --steps=20000 \ --output_dir=outputs/train/my_smolvla \ --job_name=my_smolvla_training \ --policy.device=cuda \ --wandb.enable=true
# Run the policy using the record function
python -m lerobot.record \
--robot.type=so101_follower \
--robot.port=/dev/ttyACM0 \ # <- Use your port
--robot.id=my_blue_follower_arm \ # <- Use your robot id
--robot.cameras="{ front: {type: opencv, index_or_path: 8, width: 640, height: 480, fps: 30}}" \ # <- Use your cameras
--dataset.single_task="Grasp a lego block and put it in the bin." \ # <- Use the same task description you used in your dataset recording
--dataset.repo_id=HF_USER/dataset_name \ # <- This will be the dataset name on HF Hub
--dataset.episode_time_s=50 \
--dataset.num_episodes=10 \
--policy.path=makermods/smolvla_finetune_arrange_chocolate5,000-step fine-tune of the 20k checkpoint of smolvla_3cam_50ep_arrange_chocolate, on makermods/arrange_chocolate_finetune.
| Policy | SmolVLA (smolvla) |
| Base | /home/makermods/.cache/huggingface/lerobot/outputs/train/smolvla_arrange_chocolate_20260827_204344/run/checkpoints/020000/pretrained_model |
| Dataset | makermods/arrange_chocolate_finetune |
| Steps | 5000 |
| Batch size | 64 |
| LR | 1e-05 -> 2.5e-06 (cosine_decay_with_warmup) |
| Warmup steps | 200 |
| Cameras | front, top, wrist (3x 480x640) |
| Checkpoints | 10 (every 500 steps) |
The final checkpoint is at the repo root, so this repo loads directly as a policy.
Every intermediate checkpoint is under checkpoints/<step>/, each with
pretrained_model/ and training_state/ (resumable).
rename_map is empty and the camera keys are the real dataset keys
(observation.images.front / .top / .wrist), so lerobot-eval,
lerobot-rollout and policy_server.py load this checkpoint without a rewrite.