在一些教程中,我能够使用python TensorFlow API训练模型。 我还设法冻结了最新检查点的模型。
但是,我希望使用最好的而不是最新的型号,所以我检查了将最好的型号写入光盘。最后,我以“保存的模型”格式找到了它,现在我想将其转换为冻结的模型。我认为这应该是直截了当的。
这可能是一个愚蠢的问题,但是我已经花了几个小时寻找一种简单的方法。
非常感谢您的帮助!
答案 0 :(得分:0)
保存模型后,即完成了冻结。它的重量不会改变。通常,在测试过程中,您将实例化模型,然后加载保存的即冻结的模型并对其进行预测。
如果您不希望网络在加载冻结的模型权重后学习,请转到model.layers
并为每个图层设置trainable
为false。
model.load_weights(checkpoint_path)
用于加载权重。