查看: 1347|回复: 0

tensorflow错误解决

[复制链接]

665

主题

1234

帖子

6568

积分

xdtech

Rank: 5Rank: 5

积分
6568
发表于 2019-11-25 22:29:55 | 显示全部楼层 |阅读模式
首先陈述下遇到的问题,程序中需要同时用到两个RNN模型,而且这两个RNN模型需要同时参与训练。结果出现了报错:Variable rnn/basic_lstm_cell/kernel already exists, disallowed....


再者表达下我从网上查找这类错误的经历,在google中快搜上述错误,确实会搜到一些解决办法,主流办法主要有两种:


(1)通过 with tf.variable_scope('scope',reuse=True),给不同的RNN模型定义不同的作用域
def RNN_u():
        pass
def RNN_i():
        pass
               
with tf.variable_scope('user'):
        self.h_pool_flat_u = RNN_u(self.embedded_user,weights,biases) #RNN模型开始执行的代码
with tf.variable_scope('item'):
        self.h_pool_flat_i = RNN_i(self.embedded_item,weights,biases)




(2)在代码中写上 tf.reset_default_graph()
在代码的最开始处,写入。
def run_with_input(inputs_batch, seq_lenth_batch):
    tf.reset_default_graph()

    X = tf.placeholder(tf.float32, [None, max_seq, dim])
    seq_length = tf.placeholder(tf.int32, [None])







回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表