- 这是Keras本身的一个小bug
- 如果使用时,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