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

fix config of AttentionCluster/LSTM

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