我使用tf.train.Saver('./model.ckpt')保存了模型,当我转到本地目录时,我发现了名为model.ckpt.indexmodel.ckpt.metamodel.ckpt.data-00000-of-00001 ,但不是model.ckpt 结果,我无法还原模型。 有人知道我做错了什么吗? 这是我的代码

class autoencoder(object):
    def __init__(self, network_architecture, learning_rate=0.001, regularization_constant=1):
        self.network_arch = network_architecture
        self.X = tf.placeholder(tf.float32, [None, network_architecture['n_input']])
        self.c = tf.Variable(regularization_constant, dtype=tf.float32)

        self._initialize_weights()
        self._build_graph()

        self.cost, self.optimizer = self._cost_optimizer(learning_rate)

        init = tf.global_variables_initializer()
        self.saver = tf.train.Saver()

        self.sess = tf.Session()
        self.sess.run(init)

   ...

    def save(self, path):
        self.saver.save(self.sess, path)

    def load(self, path):
        self.saver.restore(self.sess, path)

#1楼 票数:2 已采纳

* .meta文件包含您的MetaGraph,您可以通过以下方式导入它:

saver = tf.train.import_meta_graph("model.ckpt.meta")

然后,您可以还原图形的变量。

saver.restore(sess, "model.ckpt")

您可以通过调用tf.train.export_meta_graph将其他模型数据保存到tf.train.export_meta_graph

另外,您可以使用SavedModel存储应用程序的模型,该模型可以包含多个MetaGraph。 文档在这里

  ask by Chester Cheng translate from so

未解决问题?本站智能推荐:

2回复

Tensorflow、tf.reshape导致“变量不存在梯度”

我正在使用 Tensorflow/Keras(Windows 上的 TF 版本 2.1,Python 3.7)编写一个完全连接的层,但我发现如果我在乘以它之前重塑我的权重张量,那么 Tensorflow 似乎无法计算即使我只是将渐变重塑为自己的形状。 考虑以下层代码: 如果这与cause_error
1回复

OSError:SavedModel文件不存在tflite

我正在尝试将我保存的模型转换为 tflite 模型,保存的模型保存在我的桌面上,但是当我尝试运行此代码时:我生成一个错误 - 不确定是什么问题。
1回复

资源localhost/total/N10tensorflow3VarE不存在

我正在与 Google Colab 合作并尝试使用 VGG 块训练模型。 像这样: 但是,当我尝试使用 model.fit_generator 时,它给了我一个错误。 我正在使用的代码是: 我已经尝试了一切,但我不知道该怎么做。 它给了我以下错误: 我将不胜感激任何帮助。 我是新来的。 我能做什么?
1回复

OSError:当我尝试在Heroku上部署Flask应用程序时,SavedModel文件不存在

我的flask应用程序在本地服务器上运行良好。 当我尝试在heroku上部署它时,出现以下错误: 2020-11-12T13:22:11.503563+00:00 app[web.1]: OSError: SavedModel file does not exist at: /Users/leyla
1回复

SavedModel文件不存在于:model.h5/{saved_model.pbtxt|saved_model.pb}

Tensorflow 2.0 版已在 Colab 和本地试用 Ubuntu 18.4 model.h5 在当前工作目录中 输出
2回复

OSError:SavedModel文件不存在于:C:\\Users\\Munib\\Newfolder/{saved_model.pbtxt|saved_model.pb}

我想在 android studio 中使用我的 keras 训练模型。 我在互联网上得到了这段代码,将我的代码从 keras 转换为 tensorflow-lite。 但是当我尝试代码时,我收到了这个错误: 我从 keras 转换为 tensorflow-lite 时使用的代码:
5回复

TensorflowGradientTape间歇性地“变量不存在梯度”

在训练我的网络时,我偶尔会遇到警告: W0722 11:47:35.101842 140641577297728 optimizer_v2.py:928] Gradients does not exist for variables ['model/conv1d_x/Variable:0'] whe
1回复

将从tensorflow.keras.preprocessing.text_dataset_from_directory()获得的数据集保存在外部文件中

我可以将从tensorflow.keras.preprocessing.text_dataset_from_directory()获得的数据集保存在外部文件中吗? 我正在阅读文档,但不确定是否可行。 回答@Lescurel 问题我想这样做是因为我想避免每次都进行这种预处理并且必须等待它完成。 此外,