使用EfficientNetB0模型训练，出现TypeError: Unable to serialize [2.0896919 2.1128857 2.1081853] to JSON的错误
·
1 错误详情：
2 初次解决发现：
- 错误在Tensorflow==2.10.0 / 2.13.0 / 2.9.0 等环境中都无法避免
- 代码中使用ModelCheckpoint、TensorBoard等内容，会导致错误出现
3 最终解决办法：
- 修改模型 EfficientNetB0 为 EfficientNetV2B0
- 添加参数 save_weights_only=True
4 参考文献：
https://blog.csdn.net/aaaq_/article/details/127929058
https://github.com/mrdbourke/tensorflow-deep-learning/discussions/575
5 总结：
基于上述两个解决方法，博主的代码最终都可以正常训练，环境是Tensorflow==2.10.0，以上方法仅供参考，具体还需根据自身的代码和环境考虑。
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
4
0- 0
已为社区贡献1条内容
所有评论(0)