我使用 Python 2.7 和 tensorflow 1.9 和 keras 2.1.6 来训练模型并输出.h5 权重文件,但随后我想将其保存为 ckpt,稍后我将加载并用于分类:sa...
我使用 Python 2.7 和 tensorflow 1.9 和 keras 2.1.6 来训练模型并输出 .h5 权重文件,但随后我想将其保存为 ckpt,稍后我将加载并用于分类:
saver = tf.train.Saver(tracking_instance._net.trainable_collection)
model = keras.models.load_model("lenet_mnist.h5")
save_path = saver.save(sess, "/lenet_model_to_use.ckpt")
模型加载到分配的变量中,但随后崩溃并出现以下错误 saver.save()
:
tensorflow.python.framework.errors_impl.FailedPreconditionError: Attempting to use uninitialized value conv1/biases
我检查了这里并建议使用以下初始化:
sess = tf.Session()
init_op = tf.global_variables_initializer()
sess.run(init_op)
但仍然会出现相同的异常。
任何想法?
尝试将 .h5 转换为 .ckpt
下载声明:
本站所有软件和资料均为软件作者提供或网友推荐发布而来,仅供学习和研究使用,不得用于任何商业用途。如本站不慎侵犯你的版权请联系我,我将及时处理,并撤下相关内容!