KeyError: 'IteratorV2' при восстановлении модели тензорного потока - PullRequest
0 голосов
/ 02 апреля 2019

Я обучил свою модель seq2seq на google colab, работающем с tenorflow 1.13.1, с ускорением gnu, затем я пытаюсь восстановить модель на локальной машине, работающей с tenorflow 1.9.0 на процессоре в jupyter. Когда я восстанавливаю свой граф модели, делаю:

loader = tf.train.import_meta_graph(model_graph_path)

Я получаю ошибку:

KeyError: 'IteratorV2'.

Это проблема из-за разных версий tenorflow? потому что, если я полностью тренируюсь и восстанавливаю свою модель на одной машине (как на воротнике, так и на локальной машине), модель тренируется и восстанавливает нормально. Если вам нужно больше кода, чтобы понять возможную проблему, не стесняйтесь спрашивать. Заранее спасибо.

...