def build_config(self):
return Config(dataset_name='flickr8k',
epochs=None,
time_limit=timedelta(hours=10),
batch_size=32,
# As nearest as possible to 1.0, but must not be >= 1.0
reduce_lr_factor=1.0 - 1e-6,
reduce_lr_patience=sys.maxsize,
early_stopping_patience=sys.maxsize,
lemmatize_caption=True,
rare_words_handling='nothing',
words_min_occur=1,
learning_rate=0.001,
vocab_size=None,
embedding_size=300,
rnn_output_size=256,
dropout_rate=0.3,
bidirectional_rnn=False,
rnn_type='lstm',
rnn_layers=1,
l1_reg=0.0,
l2_reg=0.0,
initializer='glorot_uniform',
word_vector_init=None,
image_augmentation=False)
评论列表
文章目录