File size: 1,024 Bytes
7feb7ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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 |