tensorflow - 将模型另存为 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")

如果有人可以向我解释，我会非常高兴，我现在尝试了几个小时没有成功。

标签： tensorflowkerasconv-neural-networkdataformatstyle-transfer

Answer 1

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

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

您可以在此处阅读有关保存权重或模型和切点的更多信息