Merge pull request #1726 from heavengate/video_classification

fix config of AttentionCluster/LSTM

fluid/PaddleCV/video/configs/attention_cluster.txt

@@ -15,17 +15,17 @@ topk = 20
 epoch = 5
 learning_rate = 0.001
 pretrain_base = None
-batch_size = 160
+batch_size = 2048
 use_gpu = True
-num_gpus = 4
+num_gpus = 8
 filelist = "dataset/youtube8m/train.list"
 
 [VALID]
-batch_size = 160
+batch_size = 2048
 filelist = "dataset/youtube8m/val.list"
 
 [TEST]
-batch_size = 40
+batch_size = 256
 filelist = "dataset/youtube8m/test.list"
 
 [INFER]

fluid/PaddleCV/video/configs/attention_lstm.txt

@@ -19,17 +19,17 @@ decay_gamma = 0.1
 weight_decay = 0.0008
 num_samples = 5000000
 pretrain_base = None
-batch_size = 160
+batch_size = 1024
 use_gpu = True
-num_gpus = 4
+num_gpus = 8
 filelist = "dataset/youtube8m/train.list"
 
 [VALID]
-batch_size = 160
+batch_size = 1024
 filelist = "dataset/youtube8m/val.list"
 
 [TEST]
-batch_size = 40
+batch_size = 128
 filelist = "dataset/youtube8m/test.list"
 
 [INFER]

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