tf2.X中将模型保存为savedmodel

背景

将一个计算欧式距离的小模型转换为savedmodel。

代码

import tensorflow as tfclass EuclideanDisTanceNet(tf.keras.Model):def __init__(self):super(EuclideanDisTanceNet, self).__init__()def call(self, input):sub = tf.subtract(input[0], input[1])print('sub>', sub)pow = tf.pow(sub, 2)print('pow>', pow)rd = tf.reduce_sum(pow, axis=3)print('rd>', rd)sq = tf.math.sqrt(rd)return sqinput1 = tf.constant(3, dtype=tf.float32, shape=[1, 1, 500, 512])
input2 = tf.constant(1, dtype=tf.float32, shape=[1, 1, 1, 512])model = EuclideanDisTanceNet()# out = model([input1, input2])
# print("out>", out)out = model.predict([input1, input2])
print("out>", out)
model.save("./savedmodle", include_optimizer=False)

特别注意

如果没有调用predict()，则上面的代码会报错：

cannot be saved because the input shapes have not been set. Usually, input shapes are automatically determined from calling `.fit()` or `.predict()`. To manually set the shapes, call `model.build(input_shape)`.

参考资料

在 TensorFlow 2 中使用 TF Hub 中的 SavedModel

tensorflow2.x:构建tf.keras.Model实例的几种方式

tf2.X中将模型保存为savedmodel相关推荐

Tensorflow 模型保存与恢复（2）使用SavedModel
使用SavedModel 保存和恢复模型本篇介绍使用SavedModel进行模型的保存与恢复. 其他相关: Tensorflow 模型保存与恢复(1)使用tf.train.Saver() Tenso ...
TensorFlow模型保存和加载方法
TensorFlow模型保存和加载方法模型保存 import tensorflow as tfw1 = tf.Variable(tf.constant(2.0, shape=[1]), name=& ...
keras保存模型_TF2 8.模型保存与加载
举个例子:先训练出一个模型 import 接下来第一种方法:只保留模型的参数:这个有2种方法: model.save_weights("adasd.h5")model.load_w ...
TensorFlow2.0 —— 模型保存与加载
目录 1.Keras版本模型保存与加载 2.自定义版本模型保存与加载 3.总结 1.Keras版本模型保存与加载保存模型权重(model.save_weights) 保存HDF5文件(model.s ...
keras笔记-模型保存以及tensorboard的使用
对于一个epoch的模型保存以及tensorboard的使用都是可以使用keras中的回调函数 Callbacks. 首先还是看代码吧: from keras import backend as K ...
tensorflow的early stopping模型保存方式
自定义模型只保存参数(model_weight模型),官方模型保存参数和权重(.h5模型) 1.官方模型 model.save('save_model.h5') new_model = keras.m ...
Matlab中将数据保存为txt或dat格式四种方法
转载于 Tsingke 老师的文章--Matlab中将数据保存为txt或dat格式四种方案 - Tsingke - 博客园. ...
Tensorflow 2.x(keras)源码详解之第十章：keras中的模型保存与加载(详解Checkpointmd5模型序列化)
大家好,我是爱编程的喵喵.双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中.从事机器学习以及相关的前后端开发工作.曾在阿里云.科大讯飞.CCF等比赛获得多次Top名次.现 ...
tensor和模型保存与加载 PyTorch
PyTorch教程-7:PyTorch中保存与加载tensor和模型详解保存和读取Tensor PyTorch中的tensor可以保存成 .pt 或者 .pth 格式的文件,使用torch.save ...

tf2.X中将模型保存为savedmodel

背景

代码

特别注意

参考资料

tf2.X中将模型保存为savedmodel相关推荐

最新文章

热门文章