如何通过tf.keras.Model返回值完全保存模型并再次还原

时间:2019-07-15 04:10:10

标签: python tensorflow keras

我正在尝试使用model.save('model.h5')保存模型,但是会引发错误。查看由函数创建的def transformer:模型,其返回值为tf.keras.Model(...)

它抛出

  

TypeError :(“非JSON可序列化:”,b'\ n ...')

https://colab.research.google.com/github/tensorflow/examples/blob/master/community/en/transformer_chatbot.ipynb#scrollTo=-9Oibz2es-qW

1 个答案:

答案 0 :(得分:0)

使用pickle将模型对象作为文件转储