1 错误详情:
2 初次解决发现:
- 错误在Tensorflow==2.10.0 / 2.13.0 / 2.9.0 等环境中都无法避免
- 代码中使用ModelCheckpoint、TensorBoard等内容,会导致错误出现
3 最终解决办法:
- 修改模型 EfficientNetB0 为 EfficientNetV2B0
- 添加参数 save_weights_only=True
4 参考文献:
https://blog.youkuaiyun.com/aaaq_/article/details/127929058
https://github.com/mrdbourke/tensorflow-deep-learning/discussions/575
5 总结:
基于上述两个解决方法,博主的代码最终都可以正常训练,环境是Tensorflow==2.10.0,以上方法仅供参考,具体还需根据自身的代码和环境考虑。