使用with tf.variable_scope()
指定統(tǒng)一的initializer
船侧,那么在模型里面的話就self.src_embedding = tf.get_variable("src_emb", [SRC_VOCAB_SIZE, HIDDEN_SIZE])
這樣就行了,不用每一個都指定initializer
职辨。
initializer = tf.random_uniform_initializer(-0.05, 0.05)
# 定義訓練用的循環(huán)神經(jīng)網(wǎng)絡模型盗蟆。
with tf.variable_scope("nmt_model", reuse=None,initializer=initializer):
train_model = NMTModel()