train_tsm.sh 176 字节
Newer Older
D
dengkaipeng 已提交
1 2 3
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
python train.py --model_name="TSM" --config=./configs/tsm.txt --epoch_num=65 \
                --valid_interval=1 --log_interval=10