model = tf.keras.Model(...) # 你的model,例如 model = VGG(),这个是训练时候定义的
tf.saved_model.save(model, path) # path是你要保存的路径,例如 'model'
""" 这里是用法,也就是推理时用的 """
imported = tf.saved_model.load(path) # path代表训练后保存的模型位置
outputs = imported(inputs) # 模型输出
TF2训练与保存模型
最新推荐文章于 2023-04-13 12:04:00 发布