TF2训练与保存模型

本文主要是介绍TF2训练与保存模型，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

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训练与保存模型的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---