add visible devices in train_nextvlad.sh

75ae8caa · dengkaipeng · 27e35819 · 75ae8caa
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

fluid/PaddleCV/video/scripts/train/train_nextvlad.sh fluid/PaddleCV/video/scripts/train/train_nextvlad.sh +1 -0

未找到文件。
--- a/fluid/PaddleCV/video/scripts/train/train_nextvlad.sh
+++ b/fluid/PaddleCV/video/scripts/train/train_nextvlad.sh
+export CUDA_VISIBLE_DEVICES=0,1,2,3
 python train.py --model-name="NEXTVLAD" --config=./configs/nextvlad.txt --epoch-num=6 \
                --valid-interval=1 --log-interval=10