From 75ae8caaf55c3badb4a5798dd881bb1a65aeb3ed Mon Sep 17 00:00:00 2001
From: dengkaipeng <dengkaipeng@baidu.com>
Date: Fri, 1 Feb 2019 11:44:43 +0800
Subject: [PATCH] add visible devices in train_nextvlad.sh

---
 fluid/PaddleCV/video/scripts/train/train_nextvlad.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/fluid/PaddleCV/video/scripts/train/train_nextvlad.sh b/fluid/PaddleCV/video/scripts/train/train_nextvlad.sh
index 7aa50f72..b5857e9f 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
-- 
GitLab