提交 abb8b435 编写于 作者: S SunGaofeng

change config to match dataset rename

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