我今天在训练风格转移神经网络时遇到了保存格式的大麻烦。
我觉得任务已经解决了，我只需要保存我的模型并再次加载它。但我找不到合适的方法。
我使用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"(
如果有人能向我解释，我会非常高兴，我试了几个小时都没有成功。
您可以使用以检查点格式保存权重
model.save_weights("modelcheckpoint",save_format="tf")
你可以在这里阅读更多关于节省重量或模型和检查点的信息