tensorflow RNN实现

作者: 时间
发布时间: 2025-01-17 10:24:12 (15天前)
转自：

2 条回复

0#
回复此人
誓言你好 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 你需要使用 <code> reuse </code> 选项正确。以下更改将解决它。对于预测，您需要使用图中已存在的变量。 </p> <pre> <code> def RNN(inputs, reuse): with tf.variable_scope('cells', reuse=reuse): basic_cell = tf.contrib.rnn.BasicRNNCell(num_units=batch_size, reuse=reuse) ... ... #Training cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(labels=train_label_batch, logits=RNN(train_batch, reuse=None))) #Accuracy ... correct_prediction = tf.equal(tf.argmax(RNN(test_image, reuse=True), 1), tf.argmax(test_image_label, 0)) </code> </pre> </DIV>

编辑

登录后才能参与评论