提交 4457c08c 编写于 作者: D dengkaipeng

remove epoch/bs in train example.

上级 c666eb74
...@@ -38,8 +38,6 @@ ...@@ -38,8 +38,6 @@
export CUDA_VISIBLE_DEVICES=0 export CUDA_VISIBLE_DEVICES=0
python train.py --model-name=AttentionCluster python train.py --model-name=AttentionCluster
--config=./configs/attention_cluster.txt --config=./configs/attention_cluster.txt
--epoch=8
--batch-size=256
--save-dir=checkpoints --save-dir=checkpoints
--log-interval=10 --log-interval=10
--valid-interval=1 --valid-interval=1
...@@ -51,14 +49,18 @@ python train.py --model-name=AttentionCluster ...@@ -51,14 +49,18 @@ python train.py --model-name=AttentionCluster
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
python train.py --model-name=AttentionCluster python train.py --model-name=AttentionCluster
--config=./configs/attention_cluster.txt --config=./configs/attention_cluster.txt
--epoch=8
--batch-size=2048
--save-dir=checkpoints --save-dir=checkpoints
--log-interval=10 --log-interval=10
--valid-interval=1 --valid-interval=1
``` ```
- 请根据`CUDA_VISIBLE_DEVICES`指定卡数修改`config`文件中`num_gpus``batch_size` 视频库同时提供了快速训练脚本,脚本位于`scripts/train`目录下,可通过如下命令启动训练
``` bash
sh scripts/train/train_attention_cluster.sh
```
- 请根据`CUDA_VISIBLE_DEVICES`指定卡数修改`config`文件中的`num_gpus``batch_size`配置。
## Model Zoo ## Model Zoo
......
...@@ -34,7 +34,6 @@ Attention Cluster模型使用2nd-Youtube-8M数据集, 数据下载及准备请 ...@@ -34,7 +34,6 @@ Attention Cluster模型使用2nd-Youtube-8M数据集, 数据下载及准备请
python train.py --model-name=AttentionCluster python train.py --model-name=AttentionCluster
--config=./configs/attention_cluster.txt --config=./configs/attention_cluster.txt
--save-dir=checkpoints --save-dir=checkpoints
--epoch=8
--log-interval=10 --log-interval=10
--valid-interval=1 --valid-interval=1
...@@ -58,7 +57,8 @@ Attention Cluster模型使用2nd-Youtube-8M数据集, 数据下载及准备请 ...@@ -58,7 +57,8 @@ Attention Cluster模型使用2nd-Youtube-8M数据集, 数据下载及准备请
python test.py --model-name=AttentionCluster python test.py --model-name=AttentionCluster
--config=configs/attention_cluster.txt --config=configs/attention_cluster.txt
--log-interval=1 --weights=$PATH_TO_WEIGHTS --log-interval=1
--weights=$PATH_TO_WEIGHTS
bash scripts/test/test_attention_cluster.sh bash scripts/test/test_attention_cluster.sh
......
...@@ -30,7 +30,6 @@ AttentionLSTM模型使用2nd-Youtube-8M数据集,关于数据本分请参考[ ...@@ -30,7 +30,6 @@ AttentionLSTM模型使用2nd-Youtube-8M数据集,关于数据本分请参考[
python train.py --model-name=AttentionLSTM python train.py --model-name=AttentionLSTM
--config=./configs/attention_lstm.txt --config=./configs/attention_lstm.txt
--save-dir=checkpoints --save-dir=checkpoints
--epoch=10
--log-interval=10 --log-interval=10
--valid-interval=1 --valid-interval=1
...@@ -46,7 +45,8 @@ AttentionLSTM模型使用2nd-Youtube-8M数据集,关于数据本分请参考[ ...@@ -46,7 +45,8 @@ AttentionLSTM模型使用2nd-Youtube-8M数据集,关于数据本分请参考[
python test.py --model-name=AttentionLSTM python test.py --model-name=AttentionLSTM
--config=configs/attention_lstm.txt --config=configs/attention_lstm.txt
--log-interval=1 --weights=$PATH_TO_WEIGHTS --log-interval=1
--weights=$PATH_TO_WEIGHTS
bash scripts/test/test_attention_lstm.sh bash scripts/test/test_attention_lstm.sh
......
...@@ -34,7 +34,6 @@ StNet的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集 ...@@ -34,7 +34,6 @@ StNet的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集
python train.py --model-name=STNET python train.py --model-name=STNET
--config=./configs/attention_stnet.txt --config=./configs/attention_stnet.txt
--save-dir=checkpoints --save-dir=checkpoints
--epoch=60
--log-interval=10 --log-interval=10
--valid-interval=1 --valid-interval=1
...@@ -64,7 +63,8 @@ StNet的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集 ...@@ -64,7 +63,8 @@ StNet的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集
python test.py --model-name=STNET python test.py --model-name=STNET
--config=configs/attention_stnet.txt --config=configs/attention_stnet.txt
--log-interval=1 --weights=$PATH_TO_WEIGHTS --log-interval=1
--weights=$PATH_TO_WEIGHTS
bash scripts/test/test_attention_stnet.sh bash scripts/test/test_attention_stnet.sh
......
...@@ -29,7 +29,6 @@ TSN的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集。 ...@@ -29,7 +29,6 @@ TSN的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集。
python train.py --model-name=TSNET python train.py --model-name=TSNET
--config=./configs/attention_tsn.txt --config=./configs/attention_tsn.txt
--save-dir=checkpoints --save-dir=checkpoints
--epoch=45
--log-interval=10 --log-interval=10
--valid-interval=1 --valid-interval=1
...@@ -49,7 +48,8 @@ TSN的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集。 ...@@ -49,7 +48,8 @@ TSN的训练数据采用由DeepMind公布的Kinetics-400动作识别数据集。
python test.py --model-name=TSN python test.py --model-name=TSN
--config=configs/attention_tsn.txt --config=configs/attention_tsn.txt
--log-interval=1 --weights=$PATH_TO_WEIGHTS --log-interval=1
--weights=$PATH_TO_WEIGHTS
bash scripts/test/test_attention_tsn.sh bash scripts/test/test_attention_tsn.sh
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册