如何将检查点转换为 Keras .h5 模型？

Question

我有一个保存检查点的张量流模型，但我需要加载权重并保存 Kereas .h5 模型。我怎样才能做到这一点？

Answer 1

我假设您需要将以前的检查点转换为 .h5

给定一个已经训练好的模型，您想要加载其权重并保存为.h5。我假设您已将其保存为.model文件。假设它被称为first.model

在您的脚本中，您将需要使用 load_model，加载您的检查点

model = load_model('first.model')

那么你只需要使用

model.save('goal.h5')

另存为 .h5 文件。

为了将来参考，您可以通过将检查点保存为 .h5 来避免此转换过程：

使用检查点功能时，您可以选择另存为.model .h5或.hdf5。该行可能看起来像这样：

checkpoint = ModelCheckpoint("**FILE_NAME_HERE**.model",monitor='val_loss',verbose=1,mode='min',save_best_only=True,save_weights_only=False,period=1)

这就是将检查点保存为 .model 的方法，但将其保存为 h5 ，就像您想要做的那样：

checkpoint = ModelCheckpoint("**FILE_NAME_HERE**.h5",monitor='val_loss',verbose=1,mode='min',save_best_only=True,save_weights_only=False,period=1)