环境: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.value or 0)
在keras的源码路径下tensorflow/python/keras/_impl/keras/engine/saving.py中的get_json_type中按如上方法修改即可避免报错。
参考:https://blog.csdn.net/Funkdub/article/details/100069905