环境:win10,keras-2.2.4,tensorflow-1.13.1,python3.6
使用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.value or 0)
在keras的源码路径下tensorflow/python/keras/_impl/keras/engine/saving.py中的get_json_type中按如上方法修改即可避免报错。