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
lerobot_train_model_SmolVLA_multi_gpu_wandb.sh
Suivant →lerobot_train_model_SmolVLAmulti_gpu_wandb.sh
Articles liés
run.py
run.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/deploy/run.py).
Lire l'article →utils.py
utils.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/deploy/utils.py).
Lire l'article →agent.py
agent.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/agent.py).
Lire l'article →planner.py
planner.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/planner.py).
Lire l'article →prompts.py
prompts.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/prompts.py).
Lire l'article →run.py
run.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/run.py).
Lire l'article →