| python src/lerobot/scripts/lerobot_train.py --dataset.repo_id=Choikichang/Brick_Orange_Merged_0_99 --policy.type=diffusion --policy.repo_id=Choikichang/diffusion_brick_orange_model --output_dir=outputs/train/diffusion_brick_orange --job_name=diffusion_brick_orange_3090 --policy.device=cuda --batch_size=64 --steps=100000 --eval_freq=10000 --save_freq=10000 --wandb.enable=true | |
| PS C:\Users\user2-1\Desktop\gaussian_splatting\lerobot> python src/lerobot/scripts/lerobot_train.py ` | |
| >> --policy.pretrained_path "C:/weights" ` | |
| >> --dataset.repo_id "Choikichang/Brick_Orange_Merged_0_99" ` | |
| >> --policy.type "diffusion" ` | |
| >> --policy.repo_id "Choikichang/diffusion_brick_orange_model" ` | |
| >> --output_dir "outputs/train/diffusion_final_v5" ` | |
| >> --job_name "diffusion_3090_v5_turbo" ` | |
| >> --policy.device "cuda" ` | |
| >> --batch_size 64 ` | |
| >> --steps 100000 ` | |
| >> --eval_freq 10000 ` | |
| >> --save_freq 10000 ` | |
| >> --wandb.enable True ` | |
| >> --dataset.streaming False ` | |
| >> --num_workers 8 |