提交 abb8b435 编写于 作者: S SunGaofeng

change config to match dataset rename

上级 a931a46b
...@@ -18,16 +18,16 @@ pretrain_base = None ...@@ -18,16 +18,16 @@ pretrain_base = None
batch_size = 160 batch_size = 160
use_gpu = True use_gpu = True
num_gpus = 4 num_gpus = 4
filelist = "data/youtube8m/train.list" filelist = "dataset/youtube8m/train.list"
[VALID] [VALID]
batch_size = 160 batch_size = 160
filelist = "data/youtube8m/val.list" filelist = "dataset/youtube8m/val.list"
[TEST] [TEST]
batch_size = 40 batch_size = 40
filelist = "data/youtube8m/test.list" filelist = "dataset/youtube8m/test.list"
[INFER] [INFER]
batch_size = 1 batch_size = 1
filelist = "data/youtube8m/infer.list" filelist = "dataset/youtube8m/infer.list"
...@@ -22,16 +22,16 @@ pretrain_base = None ...@@ -22,16 +22,16 @@ pretrain_base = None
batch_size = 160 batch_size = 160
use_gpu = True use_gpu = True
num_gpus = 4 num_gpus = 4
filelist = "data/youtube8m/train.list" filelist = "dataset/youtube8m/train.list"
[VALID] [VALID]
batch_size = 160 batch_size = 160
filelist = "data/youtube8m/val.list" filelist = "dataset/youtube8m/val.list"
[TEST] [TEST]
batch_size = 40 batch_size = 40
filelist = "data/youtube8m/test.list" filelist = "dataset/youtube8m/test.list"
[INFER] [INFER]
batch_size = 1 batch_size = 1
filelist = "data/youtube8m/infer.list" filelist = "dataset/youtube8m/infer.list"
...@@ -10,7 +10,7 @@ groups = 8 ...@@ -10,7 +10,7 @@ groups = 8
expansion = 2 expansion = 2
drop_rate = 0.5 drop_rate = 0.5
gating_reduction = 8 gating_reduction = 8
eigen_file = "./data/eigenvals.npy" eigen_file = "./dataset/eigenvals.npy"
[TRAIN] [TRAIN]
epoch = 6 epoch = 6
...@@ -23,17 +23,17 @@ gradient_clip_norm = 1.0 ...@@ -23,17 +23,17 @@ gradient_clip_norm = 1.0
use_gpu = True use_gpu = True
num_gpus = 4 num_gpus = 4
batch_size = 160 batch_size = 160
filelist = "./data/youtube8m/train.list" filelist = "./dataset/youtube8m/train.list"
[VALID] [VALID]
batch_size = 160 batch_size = 160
filelist = "./data/youtube8m/val.list" filelist = "./dataset/youtube8m/val.list"
[TEST] [TEST]
batch_size = 40 batch_size = 40
filelist = "./data/youtube8m/test.list" filelist = "./dataset/youtube8m/test.list"
[INFER] [INFER]
batch_size = 40 batch_size = 40
filelist = "./data/youtube8m/infer.list" filelist = "./dataset/youtube8m/infer.list"
...@@ -17,13 +17,13 @@ buf_size = 1024 ...@@ -17,13 +17,13 @@ buf_size = 1024
batch_size = 128 batch_size = 128
num_gpus = 8 num_gpus = 8
use_gpu = True use_gpu = True
filelist = "./data/kinetics/train.list" filelist = "./dataset/kinetics/train.list"
learning_rate = 0.01 learning_rate = 0.01
learning_rate_decay = 0.1 learning_rate_decay = 0.1
l2_weight_decay = 1e-4 l2_weight_decay = 1e-4
momentum = 0.9 momentum = 0.9
total_videos = 224684 total_videos = 224684
pretrain_base = "./data/pretrained/ResNet50_pretrained" pretrain_base = "./dataset/pretrained/ResNet50_pretrained"
[VALID] [VALID]
short_size = 256 short_size = 256
...@@ -31,7 +31,7 @@ target_size = 224 ...@@ -31,7 +31,7 @@ target_size = 224
num_reader_threads = 12 num_reader_threads = 12
buf_size = 1024 buf_size = 1024
batch_size = 128 batch_size = 128
filelist = "./data/kinetics/val.list" filelist = "./dataset/kinetics/val.list"
[TEST] [TEST]
short_size = 256 short_size = 256
...@@ -39,7 +39,7 @@ target_size = 256 ...@@ -39,7 +39,7 @@ target_size = 256
num_reader_threads = 12 num_reader_threads = 12
buf_size = 1024 buf_size = 1024
batch_size = 16 batch_size = 16
filelist = "./data/kinetics/test.list" filelist = "./dataset/kinetics/test.list"
[INFER] [INFER]
short_size = 256 short_size = 256
...@@ -47,5 +47,5 @@ target_size = 256 ...@@ -47,5 +47,5 @@ target_size = 256
num_reader_threads = 12 num_reader_threads = 12
buf_size = 1024 buf_size = 1024
batch_size = 16 batch_size = 16
filelist = "./data/kinetics/infer.list" filelist = "./dataset/kinetics/infer.list"
...@@ -17,7 +17,7 @@ buf_size = 1024 ...@@ -17,7 +17,7 @@ buf_size = 1024
batch_size = 256 batch_size = 256
use_gpu = True use_gpu = True
num_gpus = 8 num_gpus = 8
filelist = "./data/kinetics/train.list" filelist = "./dataset/kinetics/train.list"
learning_rate = 0.01 learning_rate = 0.01
learning_rate_decay = 0.1 learning_rate_decay = 0.1
l2_weight_decay = 1e-4 l2_weight_decay = 1e-4
...@@ -30,7 +30,7 @@ target_size = 224 ...@@ -30,7 +30,7 @@ target_size = 224
num_reader_threads = 12 num_reader_threads = 12
buf_size = 1024 buf_size = 1024
batch_size = 256 batch_size = 256
filelist = "./data/kinetics/val.list" filelist = "./dataset/kinetics/val.list"
[TEST] [TEST]
short_size = 256 short_size = 256
...@@ -38,7 +38,7 @@ target_size = 224 ...@@ -38,7 +38,7 @@ target_size = 224
num_reader_threads = 12 num_reader_threads = 12
buf_size = 1024 buf_size = 1024
batch_size = 32 batch_size = 32
filelist = "./data/kinetics/test.list" filelist = "./dataset/kinetics/test.list"
[INFER] [INFER]
short_size = 256 short_size = 256
...@@ -46,5 +46,5 @@ target_size = 224 ...@@ -46,5 +46,5 @@ target_size = 224
num_reader_threads = 12 num_reader_threads = 12
buf_size = 1024 buf_size = 1024
batch_size = 32 batch_size = 32
filelist = "./data/kinetics/infer.list" filelist = "./dataset/kinetics/infer.list"
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册