add changes

This commit is contained in:
jade.choghari@huggingface.co
2025-11-10 14:53:17 +00:00
parent 8d9a992953
commit 2219c29690
7 changed files with 130 additions and 7 deletions
+18
View File
@@ -0,0 +1,18 @@
accelerate launch \
--multi_gpu \
--num_processes=4 \
--mixed_precision=fp16 \
$(which lerobot-train) \
--batch_size=32 \
--save_freq=5000 \
--num_workers=32 \
--dataset.repo_id=libero_dataset \
--dataset.root=/fsx/jade_choghari/datasets/libero/ \
--policy.type=xvla \
--output_dir=/fsx/jade_choghari/outputs/train/xvla_libero_multi \
--job_name=xvla_libero \
--policy.device=cuda \
--policy.action_mode=franka_joint7 \
--wandb.enable=true \
--policy.repo_id=jadechoghari/X-VLA-Libero \
--steps=10000