Views
No views yet
!python src/lerobot/scripts/lerobot_train.py \
--dataset.repo_id=aadarshram/metaworld-reach-v3 \
--policy.type=act \
--output_dir=outputs/train/act_metaworld-reach-v3 \
--job_name=act_metaworld-reach-v3 \
--policy.device=cuda \
--steps=10000 \
--batch_size=8 \
--save_freq=2000 \
--eval_freq=1000 \
--log_freq=100 \
--policy.optimizer_lr=1e-4 \
--wandb.enable=true \
--policy.repo_id=aadarshram/act_metaworld-reach-v3 \
--policy.push_to_hub=true1lerobot-train \
2 --dataset.repo_id=${HF_USER}/<dataset> \
3 --policy.type=act \
4 --output_dir=outputs/train/<desired_policy_repo_id> \
5 --job_name=lerobot_training \
6 --policy.device=cuda \
7 --policy.repo_id=${HF_USER}/<desired_policy_repo_id>
8 --wandb.enable=trueoutputs/train/<desired_policy_repo_id>/checkpoints/.1lerobot-record \
2 --robot.type=so100_follower \
3 --dataset.repo_id=<hf_user>/eval_<dataset> \
4 --policy.path=<hf_user>/<desired_policy_repo_id> \
5 --episodes=10--policy.path pointing to a local or hub checkpoint.