我今天在训练风格转移神经网络时保存格式遇到了很大的麻烦。
我觉得任务已经解决了,我只需要保存我的模型并再次加载它。但我找不到合适的方法来做到这一点。
我使用来自 github 的以下代码来训练风格转移网络:
我已经成功地训练了网络。
现在,我使用以下行保存了模型:
model.save("/tmp/nst/test.h5")
不过,为了应用保存的神经网络,我需要使用 .ckpt 格式的网络。
有人能告诉我如何在 h5 和 .ckpt 之间切换数据格式吗?
或者是否有针对keras的特定保存方法,所以我可以将其保存为.ckpt?(--> 伪代码:model.save_cpkt("/tmp/nst/test.ckpt")
如果有人可以向我解释,我会非常高兴,我现在尝试了几个小时没有成功。