Update train_mvtec.sh

CASIA-IVA-Lab · Oct 11, 2023 · bcaad14 · bcaad14
1 parent f545288
commit bcaad14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/scripts/train_mvtec.sh b/code/scripts/train_mvtec.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-deepspeed --include localhost:5,7 --master_port 28400 train_mvtec.py \
+deepspeed --include localhost:0,1 --master_port 28400 train_mvtec.py \
     --model openllama_peft \
     --stage 1\
     --imagebind_ckpt_path ../pretrained_ckpt/imagebind_ckpt/imagebind_huge.pth\