diff --git a/fluid/PaddleCV/video/scripts/train/train_nextvlad.sh b/fluid/PaddleCV/video/scripts/train/train_nextvlad.sh index 7aa50f7298c038c622c32d1e168eb2168057bcee..b5857e9f35a47d89ce5185c6d42b2ed51207e390 100644 --- a/fluid/PaddleCV/video/scripts/train/train_nextvlad.sh +++ b/fluid/PaddleCV/video/scripts/train/train_nextvlad.sh @@ -1,2 +1,3 @@ +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