S SmartDocs
Serie: Humanoid bash 13 righe · Aggiornato 2026-03-12

lerobot_train_model_SmolVLA_wandb.sh

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

export CUDA_VISIBLE_DEVICES=0
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True

lerobot-train \
  --policy.path=lerobot/smolvla_base \
  --policy.repo_id=${HF_USER}/lerobot_pick_and_place_smolvla_rgbd \
  --dataset.repo_id=${HF_USER}/pick_place_with_rgbd \
  --batch_size=16 \
  --steps=20000 \
  --output_dir=outputs/train/my_smolvla \
  --job_name=pick_and_place_SmolVLA \
  --policy.device=cuda \
  --wandb.enable=true

Articoli correlati