Keras问题：Save Model: TypeError:(‘Not JSON Serializable:‘, Dimension(3))

鱼吐泡泡水

于 2020-09-14 12:46:31 发布

阅读量898

点赞数 2

分类专栏： tips 文章标签： tensorflow

本文链接：https://blog.youkuaiyun.com/m0_37859875/article/details/108577165

版权

在使用Keras 2.2.2与TensorFlow 2.1.0环境下，遇到模型保存时出现的bug。通过修改Keras源码中tensorflow/python/keras/_impl/keras/engine/saving.py文件的get_json_type方法，可以有效解决此问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

环境：keras-2.2.2，tensorflow-2.1.0，python3.7

使用keras训练时保存模型的时候报错，解决方案比较扯淡- -貌似是一个不愿意解决的bug

这个需改keras的源码，具体的改正方法见github的issues9342 https://github.com/keras-team/keras/issues/9342

from tensorflow.python.framework.tensor_shape import Dimension
if type(obj) == Dimension:
	return int(obj