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

Question

Я использую Google Colab Laboratory для реализации глубокого обучения в Python3. Я создаю модель, обучаю ее, испытываю. Все отлично. Наконец я пытаюсь сохранить модель на моем диске Google. Но это говорит Ошибка: в настоящее время для сохранения требуется, чтобы модель была графовой сетью.

До тренировок и испытаний проблем нет.

Затем я монтирую диск

from google.colab import drive
drive.mount('/content/gdrive')

А затем попытайтесь сохранить модель для последующего использования как:

model.save('my_model_name.model')

Но это не сохранение модели. Чего мне не хватает?

Answer 1

предпочтительный способ сохранить модель с тензорным потоком - использовать модуль tf.train.Saver(). Допустим, ваша модель называется просто Модель , и вы хотите сохранить ее в определенном каталоге. Это предпочтительный способ сделать это.

import tensorflow as tf
directory_to_save = '/content/drive'
with tf.Session() as sess
    saver = tf.train.Saver()
    #train model
    saver.save(sess, directory_to_save)