Проблемы восстановления модели Tensorflow - PullRequest
0 голосов
/ 13 сентября 2018

Я продолжаю находить следующую проблему во время восстановления модели RNN в Tensorflow.Модель работает хорошо, но у восстановления есть проблема.ниже мой код.В чем может быть проблема.Я удалил и очистил всю предыдущую сессию, но та же проблема сохраняется.

saver = tf.train.import_meta_graph(os.path.join(logs_path, 
'Multi_Layer_RNN_model.ckpt-499999.meta'))

with tf.Session() as sess:
    ckpt = tf.train.get_checkpoint_state(logs_path)
    saver.restore(sess, tf.train.latest_checkpoint(logs_path))  

ОШИБКА

InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [105,400] rhs shape= [111,400]
     [[Node: save/Assign_11 = Assign[T=DT_FLOAT, _class=["loc:@rnn/lstm_cell/kernel"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](rnn/lstm_cell/kernel/RMSProp_1, save/RestoreV2:11)]]

Caused by op 'save/Assign_11', defined at:
  File "C:\Users\ADMN\AppData\Local\conda\conda\envs\tensorflow\lib\runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "C:\Users\ADMN\AppData\Local\conda\conda\envs\tensorflow\lib\runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "C:\Users\ADMN\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\ipykernel_launcher.py", line 16, in <module>
    app.launch_new_instance()
  File "C:\Users\ADMN\AppData\Local\conda\conda\envs\tensorflow\lib\site-packages\traitlets\config\application.py", line 658, in launch_instance
    app.start()
  .................

1 Ответ

0 голосов
/ 13 сентября 2018

В первой строке у вас есть причина:

lhs shape= [105,400] rhs shape= [111,400]

Кажется, что у вас есть разные размеры вектора / тензора между нейронными сетями.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...