lerobot_train_model_wandb.sh
Humanoid/fmc3-robotics-main/projects/lerobot/demo_scripts/lerobot_train_model_wandb.sh
export CUDA_VISIBLE_DEVICES=1
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
lerobot-train \
--dataset.repo_id=${HF_USER}/pick_place_with_rgbd \
--policy.type=act \
--output_dir=outputs/train/pick_place_with_rgbd \
--job_name=pick_and_place_act \
--policy.device=cuda \
--wandb.enable=true \
--policy.repo_id=${HF_USER}/lerobot_pick_and_place_act_rgbd
lerobot_train_model_multi_gpu_wandb_diffusion.sh
Siguiente →lerobot_train_model_wandb_diffusion.sh
Artículos relacionados
run.py
run.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/deploy/run.py).
Leer artículo →utils.py
utils.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/deploy/utils.py).
Leer artículo →agent.py
agent.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/agent.py).
Leer artículo →planner.py
planner.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/planner.py).
Leer artículo →prompts.py
prompts.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/prompts.py).
Leer artículo →run.py
run.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/run.py).
Leer artículo →