提交 1b5ce43e 编写于 作者: D dengkaipeng

add scripts for TSM.

上级 e8726492
...@@ -31,7 +31,7 @@ TSM的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集。 ...@@ -31,7 +31,7 @@ TSM的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集。
--log-interval=10 --log-interval=10
--valid-interval=1 --valid-interval=1
bash scripts/train/train_tsm,.sh bash scripts/train/train_tsm.sh
- 可下载已发布模型[model](https://paddlemodels.bj.bcebos.com/video_classification/tsm_kinetics.tar.gz)通过`--resume`指定权重存放路径进行finetune等开发 - 可下载已发布模型[model](https://paddlemodels.bj.bcebos.com/video_classification/tsm_kinetics.tar.gz)通过`--resume`指定权重存放路径进行finetune等开发
......
python infer.py --model_name="TSM" --config=./configs/tsm.txt --filelist=./data/kinetics/infer.list \
--log_interval=10 --weights=./checkpoints/TSM_epoch0 --save_dir=./save
python test.py --model_name="TSM" --config=./configs/tsm.txt \
--log_interval=10 --weights=./checkpoints/TSM_epoch0
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_num=5 \
--valid_interval=1 --log_interval=10 --valid_interval=1 --log_interval=10
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_num=10 \
--valid_interval=1 --log_interval=10 --valid_interval=1 --log_interval=10
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_num=60 \
--valid_interval=1 --log_interval=10 --valid_interval=1 --log_interval=10
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
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_num=45 \
--valid_interval=1 --log_interval=10 --valid_interval=1 --log_interval=10
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册