У меня были большие проблемы с сохранением форматов при обучении нейронной сети с передачей стиля.
Я чувствую, что задача уже решена, мне нужно только сохранить мою модель и загрузить ее снова. Но я не могу найти правильный способ сделать это.
Я использовал следующий код из 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?
Или есть определенный c метод сохранения для кератов, так что я могу сохранить его как .ckpt? (-> pseudocode: model.save_cpkt ("/ tmp / nst / test.ckpt")
Был бы очень рад, если бы кто-то мог мне это объяснить, я пытался это несколько часов безуспешно.