0
点赞
收藏
分享

微信扫一扫

二次调用 tf.get_variable 例如双塔模型时解决方法

zhaoxj0217 2022-03-15 阅读 68

二次调用 tf.get_variable
出错提示:Did you mean to set reuse=tf.AUTO_REUSE in VarScope?

例如双塔模型

解决方法

embedding_table = tf.get_variable(# [vocab_size, embedding_size]
    name=word_embedding_name,
    shape=[vocab_size, embedding_size],
    initializer=create_initializer(initializer_range))

后面加一行
tf.get_variable_scope().reuse_variables()

如上所示可解决

举报

相关推荐

0 条评论