Merge pull request #2051 from heavengate/tsm_gc

gc flag for TSM.

PaddleCV/video/models/tsm/README.md

@@ -34,6 +34,9 @@ TSM的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集。
 
 数据准备完毕后，可以通过如下两种方式启动训练：
 
+    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
             --save_dir=checkpoints

PaddleCV/video/scripts/train/train_tsm.sh

 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
+
+# activate eager gc to reduce memory use
+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 \
                 --valid_interval=1 --log_interval=10