modify epoch and batch size

fluid/PaddleCV/video/configs/nextvlad.txt

@@ -34,6 +34,6 @@ batch_size = 40
 filelist = "./dataset/youtube8m/test.list"
 
 [INFER]
-batch_size = 40
+batch_size = 1
 filelist = "./dataset/youtube8m/infer.list"
 

fluid/PaddleCV/video/configs/stnet.txt

@@ -46,6 +46,6 @@ short_size = 256
 target_size = 256
 num_reader_threads = 12
 buf_size = 1024
-batch_size = 16
+batch_size = 1
 filelist = "./dataset/kinetics/infer.list"
 

fluid/PaddleCV/video/configs/tsn.txt

@@ -45,6 +45,6 @@ short_size = 256
 target_size = 224
 num_reader_threads = 12
 buf_size = 1024
-batch_size = 32
+batch_size = 1
 filelist = "./dataset/kinetics/infer.list"
 

fluid/PaddleCV/video/scripts/test/test_nextvlad.sh

-python3 test.py --model-name="NEXTVLAD" --config=./configs/nextvlad.txt \
+python test.py --model-name="NEXTVLAD" --config=./configs/nextvlad.txt \
                 --log-interval=10 --weights=./checkpoints/NEXTVLAD_epoch0

fluid/PaddleCV/video/scripts/train/train_nextvlad.sh

-python3 train.py --model-name="NEXTVLAD" --config=./configs/nextvlad.txt --epoch-num=6 \
+python train.py --model-name="NEXTVLAD" --config=./configs/nextvlad.txt --epoch-num=6 \
                 --valid-interval=1 --save-interval=1 --log-interval=10

fluid/PaddleCV/video/scripts/train/train_stnet.sh

-python3 train.py --model-name="STNET" --config=./configs/stnet.txt --epoch-num=6 \
+python3 train.py --model-name="STNET" --config=./configs/stnet.txt --epoch-num=60 \
                 --valid-interval=1 --save-interval=1 --log-interval=10

fluid/PaddleCV/video/scripts/train/train_tsn.sh

-python3 train.py --model-name="TSN" --config=./configs/tsn.txt --epoch-num=6 \
+python3 train.py --model-name="TSN" --config=./configs/tsn.txt --epoch-num=45 \
                 --valid-interval=1 --save-interval=1 --log-interval=10