Невозможно восстановить модели в tensorflow v0.8

Question

Я пытаюсь восстановить сохраненную модель. Но это возвращает мне ошибку. Пожалуйста, помогите мне. код, чтобы сохранить модель: save_model.py

import tensorflow as tf 
v1 = tf.Variable(1.32, name="v1") 
v2 = tf.Variable(1.33, name="v2") 

init = tf.initialize_all_variables() 

saver = tf.train.Saver() 

with tf.Session() as sess: 
    sess.run(init) 
    save_path = saver.save(sess, "model.ckpt") 

код для восстановления модели: restore_model.py

import tensorflow as tf 
v1 = tf.Variable(0, name="v1") 
v2 = tf.Variable(0, name="v2") 


saver = tf.train.Saver() 

with tf.Session() as sess: 
    saver.restore(sess, "model.ckpt") 
    print("Model restored.") 

Я сохранил оба файла в том же каталоге.

Answer 1

Я подозреваю, что ошибка возникает потому, что в save_model.py объявляются переменные, имеющие тип tf.float32 (неявный тип 1.32 и 1.33), в то время как в restore_model.py вы определяете переменные как имеющие тип tf.int32 (неявный тип 0).

Проще всего было бы изменить restore_model.py, чтобы объявить переменные как tf.float32. Например, вы можете сделать следующее:

v1 = tf.Variable(0.0, name="v1") 
v2 = tf.Variable(0.0, name="v2")