Как импортировать обученную глубокую модель CNN?

Question

Я обучил глубокую модель CNN на GPU в кластере с использованием Tensorflow.Теперь я хочу загрузить эту модель в свой ноутбук.Но когда я хочу использовать следующую команду, я получаю сообщение об ошибке:

new_saver = tf.train.import_meta_graph('./3/Model_Arch3/Deep_CNN_Color_Arch8.ckpt-178000.meta')

Я получаю сообщение об ошибке:

Traceback (последний вызов был последним): Файл "dev_test.py ", строка 28, в файле new_saver = tf.train.import_meta_graph ('./ 3 / Model_Arch3 / Deep_CNN_Color_Arch8.ckpt-178000.meta')" /usr/local/lib/python2.7/dist-packages/tensorflow/python/training/saver.py ", строка 1810, в файле import_meta_graph ** kwargs)" /usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/meta_graph.py ", строка 660,в файле import_scoped_meta_graph provider_op_list = provider_op_list) Файл "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/importer.py", строка 285, в import_graph_def поднимите ValueError («Не определено имя% s в определеноопераций. '% node.op) ValueError: Нет операции с именем ParseSingleExample в определенных операциях.

Я действительно ценю, если кто-то поможет мне с этой проблемой.

Answer 1

Я нашел в чем проблема.Я использовал Python 3.5.4 для обучения.Затем я использовал Python 2.7 для восстановления графа, и это было проблемой.Спасибо от тех, кто пытался помочь

Answer 2

Это также может быть проблема с отложенной загрузкой операции, которую вы пытаетесь выполнить.Попробуйте добавить dir(tf.contrib) строку перед загрузкой графика.

Для справки - аналогичная проблема возникла в старой проблеме TF GitHub https://github.com/tensorflow/tensorflow/issues/10130

---

ОБНОВЛЕНИЕ:

Я думаю, что вижу проблему сейчас.Так как вы тренировались на GPU и сейчас работаете на обычной машине / процессоре.Вам необходимо очистить устройства от загруженного объекта.Попробуйте добавить clear_devices в качестве параметра:

dir(tf.contrib)
with tf.Session() as sess:
    new_saver = tf.train.import_meta_graph(full_file_path, clear_devices=True)

Наконец, ваша версия TF между вашей локальной и графической машиной может быть не синхронизирована.Попробуйте обновить локальную версию.