Delete run_qwenlatent_vla.sh
Browse files- run_qwenlatent_vla.sh +0 -27
run_qwenlatent_vla.sh
DELETED
|
@@ -1,27 +0,0 @@
|
|
| 1 |
-
#export NCCL_SOCKET_IFNAME=bond0
|
| 2 |
-
#export NCCL_IB_HCA=mlx5_2,mlx5_3
|
| 3 |
-
|
| 4 |
-
export NCCL_BLOCKING_WAIT=1
|
| 5 |
-
export NCCL_ASYNC_ERROR_HANDLING=1
|
| 6 |
-
export NCCL_TIMEOUT=1000 # timeout set to 1 hour (unit: seconds)
|
| 7 |
-
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
|
| 8 |
-
|
| 9 |
-
# === Please modify the following paths according to your environment ===
|
| 10 |
-
###########################################################################################
|
| 11 |
-
run_root_dir=./runs
|
| 12 |
-
run_id=0418_QwenLatent_13tasks_actionstate_30k
|
| 13 |
-
###########################################################################################
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
output_dir=${run_root_dir}/${run_id}
|
| 17 |
-
mkdir -p ${output_dir}
|
| 18 |
-
# mv this script to the output dir
|
| 19 |
-
cp $0 ${output_dir}/
|
| 20 |
-
|
| 21 |
-
accelerate launch \
|
| 22 |
-
--config_file ./starVLA/config/deepseeds/deepspeed_zero2.yaml \
|
| 23 |
-
--num_processes 8 \
|
| 24 |
-
starVLA/training/train_qwenlatent.py \
|
| 25 |
-
--config_yaml ./starVLA/config/training/starvla_train_qwenlatent_oxe.yaml \
|
| 26 |
-
--run_root_dir ${run_root_dir} \
|
| 27 |
-
--run_id ${run_id} \
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|