Восстановление сохраненной модели - PullRequest
Новая
       19

Восстановление сохраненной модели

0 голосов
/ 28 апреля 2018

Я пытался предсказать рукописные цифры с помощью наборов данных tenorflow и mnist. Но я получаю ошибку. Как я могу решить ошибку? 

import tensorflow as tf
.
.    #code
.    #code
.    #code
.
y_pred= tf.nn.softmax(layer3)  # USING SOFTMAX

init_op= tf.initialize_all_variables()
saver= tf.train.Saver()

# RESTORING MODEL
with tf.Session() as sess:
    sess.restore(sess, 'model.ckpt')
    sess.run(init_op)

    prediction= tf.argmax(y_pred, 1)
    print(prediction.eval(feed_dict={x: [imvalue],keep_prob: 1.0}
                       ,session=sess))

Код выдает следующее сообщение об ошибке: 

Traceback (most recent call last):
  File "/home/monojit/writtenRecognition/digi.py", line 127, in <module>
sess.restore(sess, 'model.ckpt')
AttributeError: 'Session' object has no attribute 'restore'
...