提交 27e35819 编写于 作者: D dengkaipeng

fix config of AttentionCluster/LSTM

上级 8fb135da
......@@ -15,17 +15,17 @@ topk = 20
epoch = 5
learning_rate = 0.001
pretrain_base = None
batch_size = 160
batch_size = 2048
use_gpu = True
num_gpus = 4
num_gpus = 8
filelist = "dataset/youtube8m/train.list"
[VALID]
batch_size = 160
batch_size = 2048
filelist = "dataset/youtube8m/val.list"
[TEST]
batch_size = 40
batch_size = 256
filelist = "dataset/youtube8m/test.list"
[INFER]
......
......@@ -19,17 +19,17 @@ decay_gamma = 0.1
weight_decay = 0.0008
num_samples = 5000000
pretrain_base = None
batch_size = 160
batch_size = 1024
use_gpu = True
num_gpus = 4
num_gpus = 8
filelist = "dataset/youtube8m/train.list"
[VALID]
batch_size = 160
batch_size = 1024
filelist = "dataset/youtube8m/val.list"
[TEST]
batch_size = 40
batch_size = 128
filelist = "dataset/youtube8m/test.list"
[INFER]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册