S SmartDocs
Série: Humanoid bash 15 lignes · Mis à jour 2026-03-12

lerobot_train_model_SmolVLAmulti_gpu_wandb.sh

Humanoid/fmc3-robotics-main/projects/lerobot/demo_scripts/lerobot_train_model_SmolVLAmulti_gpu_wandb.sh

accelerate launch \
  --multi_gpu \
  --num_processes=1 \
  $(which lerobot-train) \
  --dataset.repo_id=${HF_USER}/my_eval_smolvla_single_cam  \
  --policy.path=lerobot/smolvla_base \
  --policy.repo_id=${HF_USER}/lerobot_pick_and_place_smolvla \
  --output_dir=outputs/train/act_multi_gpu/lerobot_model \
  --job_name=pick_and_place_bottle_from_A_to_B \
  --policy.device=cuda \
  --batch_size=8 \
  --steps=50000 \
  --optimizer.lr=2e-4 \
  --wandb.enable=true 
  

Articles liés