将模型另存为h5 /将模型另存为.ckpt

Question

今天，我在训练样式转换神经网络时在保存格式方面遇到了很大的麻烦。

我觉得任务已经解决，我只需要保存模型并再次加载它。但是我找不到合适的方法。

我使用了github中的以下代码来训练样式传递网络：

https://github.com/nikhilagrawal2000/Neural-Style-Transfer-with-Eager-Execution/blob/master/Neural_Style_Transfer_with_Eager_Execution.ipynb

我已经成功地训练了网络。

现在，我使用以下行保存了模型：

model.save("/tmp/nst/test.h5")

尽管要应用保存的神经网络，我需要使用.ckpt格式的网络。

有人可以告诉我如何在h5和.ckpt之间切换数据格式吗？

或者是否有用于keras的特定保存方法，所以我可以将其保存为.ckpt？ （->伪代码：model.save_cpkt（“ / tmp / nst / test.ckpt”）

如果有人可以向我解释这一点，我将非常高兴，我已经尝试了几个小时而没有成功。

Answer 1

您可以使用以下方法以检查点格式保存权重：

model.save_weights("modelcheckpoint",save_format="tf")

您可以阅读有关保存权重或模型和支点here

的更多信息