tensorflow中Modelcheckpoint无法保存h5文件及加载问题

tensorflow中Modelcheckpoint无法保存h5文件及加载问题

保存问题

保存路径一定要正确,报的错误为OSError: Unable to create link (name already exists)
则可能是自定义的网络层需要给每个权重及偏置命名,比如

self.kernel = self.add_weight(shape=kernel_shape,
                                      initializer=self.kernel_initializer,
                                      regularizer=self.kernel_regularizer,
                                      constraint=self.kernel_constraint,
                                      trainable=True,
                                      name='kernel')

中需要给weight命名为kernel,以便模型网络保存好后层的名称不会重复
该错误还会伴随错误AttributeError: ‘OSError’ object has no attribute ‘message’
也可以尝试以上方法进行破解,如果又自定义网络层,一定要保证自定义网络层的书写正确,并且参数可训练可更新。可以利用标准网络来验证自己的网络出现的错误环节
本人这些天还遇到了h5文件不能保存,或者保存后加载错误的问题,最终在卸载了已有的h5py包后重新安装pip insall h5py==2.1.0后解决该问题
关于modelcheckpoint这个坑搞了好久,最后发现还是自己的自定义网络的问题
书写优点混乱,时间仓促,就这样吧

上一篇:JVM性能分析与优化


下一篇:mysql解决乱码