fix video train param name epoch. (#2266)

* fix video train param name epoch.

* fix for script.

PaddleCV/video/scripts/train/train_attention_cluster.sh

 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
-python train.py --model_name="AttentionCluster" --config=./configs/attention_cluster.txt --epoch_num=5 \
+python train.py --model_name="AttentionCluster" --config=./configs/attention_cluster.txt --epoch=5 \
                 --valid_interval=1 --log_interval=10

PaddleCV/video/scripts/train/train_attention_lstm.sh

 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
-python train.py --model_name="AttentionLSTM" --config=./configs/attention_lstm.txt --epoch_num=10 \
+python train.py --model_name="AttentionLSTM" --config=./configs/attention_lstm.txt --epoch=10 \
                 --valid_interval=1 --log_interval=10

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 \
+python train.py --model_name="NEXTVLAD" --config=./configs/nextvlad.txt --epoch=6 \
                 --valid_interval=1 --log_interval=10

PaddleCV/video/scripts/train/train_nonlocal.sh

-python train.py --model_name="NONLOCAL" --config=./configs/nonlocal.txt --epoch_num=120 \
+python train.py --model_name="NONLOCAL" --config=./configs/nonlocal.txt --epoch=120 \
                 --valid_interval=1 --log_interval=1 \
                 --pretrain=./pretrained/ResNet50_pretrained

PaddleCV/video/scripts/train/train_stnet.sh

 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
-python train.py --model_name="STNET" --config=./configs/stnet.txt --epoch_num=60 \
+python train.py --model_name="STNET" --config=./configs/stnet.txt --epoch=60 \
                 --valid_interval=1 --log_interval=10

PaddleCV/video/scripts/train/train_tsm.sh

@@ -5,5 +5,5 @@ export FLAGS_fast_eager_deletion_mode=1
 export FLAGS_eager_delete_tensor_gb=0.0
 export FLAGS_fraction_of_gpu_memory_to_use=0.98
 
-python train.py --model_name="TSM" --config=./configs/tsm.txt --epoch_num=65 \
+python train.py --model_name="TSM" --config=./configs/tsm.txt --epoch=65 \
                 --valid_interval=1 --log_interval=10

PaddleCV/video/scripts/train/train_tsn.sh

 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
-python train.py --model_name="TSN" --config=./configs/tsn.txt --epoch_num=45 \
+python train.py --model_name="TSN" --config=./configs/tsn.txt --epoch=45 \
                 --valid_interval=1 --log_interval=10

PaddleCV/video/train.py

@@ -79,7 +79,7 @@ def parse_args():
         default=False,
         help='whether to use memory optimize in train')
     parser.add_argument(
-        '--epoch_num',
+        '--epoch',
         type=int,
         default=0,
         help='epoch number, 0 for read from config file')
@@ -197,7 +197,7 @@ def train(args):
     valid_fetch_list = [valid_loss.name] + [x.name for x in valid_outputs
                                             ] + [valid_feeds[-1].name]
 
-    epochs = args.epoch_num or train_model.epoch_num()
+    epochs = args.epoch or train_model.epoch_num()
 
     if args.no_use_pyreader:
         train_feeder = fluid.DataFeeder(place=place, feed_list=train_feeds)