错误：当前“保存”要求模型是图网络。无法保存模型答案

【问题标题】：Error: Currently `save` requires model to be a graph network. Unable to save model错误：当前“保存”要求模型是图网络。无法保存模型
【发布时间】：2019-06-02 00:42:17
【问题描述】：

我正在使用 google colaboratory 在 python3 中实现深度学习。我创建一个模型，训练它，测试它。一切顺利。最后我尝试将模型保存在我的谷歌驱动器上。但是是说 错误：当前“保存”要求模型是图网络。

直到训练和测试都没有问题。

然后我安装驱动器

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

然后尝试保存模型以供以后使用：

model.save('my_model_name.model')

但它没有保存模型。我错过了什么？

【问题讨论】：

标签： python-3.x machine-learning deep-learning google-colaboratory

【解决方案1】：

使用 tensorflow 保存模型的首选方法是使用 tf.train.Saver() 模块。因此，假设您的模型简称为 Model，并且您希望将其保存在特定目录中。这是执行此操作的首选方式。

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)

【讨论】：