美文网首页
[tf]定义默认初始化的方式,

[tf]定义默认初始化的方式,

作者: VanJordan | 来源:发表于2018-12-17 14:22 被阅读4次

    使用with tf.variable_scope()指定统一的initializer,那么在模型里面的话就self.src_embedding = tf.get_variable("src_emb", [SRC_VOCAB_SIZE, HIDDEN_SIZE])这样就行了,不用每一个都指定initializer

    initializer = tf.random_uniform_initializer(-0.05, 0.05)
    # 定义训练用的循环神经网络模型。
    with tf.variable_scope("nmt_model", reuse=None,initializer=initializer):
        train_model = NMTModel()
    

    相关文章

      网友评论

          本文标题:[tf]定义默认初始化的方式,

          本文链接:https://www.haomeiwen.com/subject/oloqkqtx.html