冻结已保存的张量流模型

时间:2020-05-13 13:59:54

标签: python tensorflow

在一些教程中,我能够使用python TensorFlow API训练模型。 我还设法冻结了最新检查点的模型。

但是,我希望使用最好的而不是最新的型号,所以我检查了将最好的型号写入光盘。最后,我以“保存的模型”格式找到了它,现在我想将其转换为冻结的模型。我认为这应该是直截了当的。

这可能是一个愚蠢的问题,但是我已经花了几个小时寻找一种简单的方法。

非常感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

保存模型后,即完成了冻结。它的重量不会改变。通常,在测试过程中,您将实例化模型,然后加载保存的即冻结的​​模型并对其进行预测。

如果您不希望网络在加载冻结的模型权重后学习,请转到model.layers并为每个图层设置trainable为false。

model.load_weights(checkpoint_path)用于加载权重。