Keras保存模型时报错“Not JSON Serializable”的解决方案

Posted chenying66

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Keras保存模型时报错“Not JSON Serializable”的解决方案相关的知识,希望对你有一定的参考价值。

  1. 这是Keras本身的一个小bug
  2. 如果使用时,Keras的backend是Tensorflow时,可以采用下述方案解决:
    在site-packages>keras>engine>saving.py文件中加上下述代码
    from tensorflow.python.framework.tensor_shape import Dimension
    if type(obj) == Dimension:
    return int(obj.value or 0)
    参考链接:
    https://github.com/keras-team/keras/issues/9342#issuecomment-396056333

以上是关于Keras保存模型时报错“Not JSON Serializable”的解决方案的主要内容,如果未能解决你的问题,请参考以下文章