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
関連記事
run.py
run.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/deploy/run.py).
記事を読む →utils.py
utils.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/deploy/utils.py).
記事を読む →agent.py
agent.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/agent.py).
記事を読む →planner.py
planner.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/planner.py).
記事を読む →prompts.py
prompts.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/agents/prompts.py).
記事を読む →run.py
run.py — python source code from the Humanoid learning materials (Humanoid/May_12/RoboOS/master/run.py).
記事を読む →