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