环境: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中按如上方法修改即可避免报错。
在Win10环境下,使用Keras 2.2.4和TensorFlow 1.13.1训练模型并保存时遇到错误。通过修改Keras源码中的saving.py文件,可以避免这一问题。具体修改涉及将Dimension类型转换为整数。
927





