Update README.md
Browse files
README.md
CHANGED
|
@@ -1 +1,19 @@
|
|
| 1 |
-
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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
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
|
| 2 |
+
|
| 3 |
+
|
| 4 |
+
|
| 5 |
+
PS C:\Users\user2-1\Desktop\gaussian_splatting\lerobot> python src/lerobot/scripts/lerobot_train.py `
|
| 6 |
+
>> --policy.pretrained_path "C:/weights" `
|
| 7 |
+
>> --dataset.repo_id "Choikichang/Brick_Orange_Merged_0_99" `
|
| 8 |
+
>> --policy.type "diffusion" `
|
| 9 |
+
>> --policy.repo_id "Choikichang/diffusion_brick_orange_model" `
|
| 10 |
+
>> --output_dir "outputs/train/diffusion_final_v5" `
|
| 11 |
+
>> --job_name "diffusion_3090_v5_turbo" `
|
| 12 |
+
>> --policy.device "cuda" `
|
| 13 |
+
>> --batch_size 64 `
|
| 14 |
+
>> --steps 100000 `
|
| 15 |
+
>> --eval_freq 10000 `
|
| 16 |
+
>> --save_freq 10000 `
|
| 17 |
+
>> --wandb.enable True `
|
| 18 |
+
>> --dataset.streaming False `
|
| 19 |
+
>> --num_workers 8
|