lerobot icon indicating copy to clipboard operation
lerobot copied to clipboard

Add the option to finetune a policy on an MPS device

Open marinabar opened this issue 1 year ago • 0 comments

What this does

Adds the option to finetune a policy on a new device (MacBook MPS).

This way the weights are loaded from the safetensors file.

The YAML config is copied from the finetuned_policy_path, which should be passed in via +finetune=pretrained_policy_path

How it was tested

python lerobot/scripts/train.py +finetune=m1b/act_koch_pick_lego_green_box

How to checkout & try? (for the reviewer)

python lerobot/scripts/train.py +finetune=lerobot/act_aloha_sim_transfer_cube_human

marinabar avatar Jul 23 '24 17:07 marinabar