changed CUDA_VISIBLE_DEVICES to gpu0

youralien · Sep 27, 2022 · 9d375f5 · 9d375f5
1 parent fe2a44d
commit 9d375f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/codes_zcj/RUN/train_strat.sh b/codes_zcj/RUN/train_strat.sh
@@ -1,4 +1,4 @@
-CUDA_VISIBLE_DEVICES=1 python train.py \
+CUDA_VISIBLE_DEVICES=0 python train.py \
     --config_name strat \
     --inputter_name strat \
     --eval_input_file ./_reformat/valid.txt \
@@ -13,4 +13,4 @@ CUDA_VISIBLE_DEVICES=1 python train.py \
     --warmup_steps 100 \
     --fp16 false \
     --loss_scale 0.0 \
-    --pbar true
+    --pbar true