Python3报错：TypeError: can't pickle _thread.RLock objects

最新推荐文章于 2025-03-26 09:17:52 发布

夜幕下的光123

最新推荐文章于 2025-03-26 09:17:52 发布

阅读量1.4w

点赞数 7

分类专栏： Python基础深度学习文章标签： Python

本文链接：https://blog.youkuaiyun.com/songchunxiao1991/article/details/84335248

版权

Python基础同时被 2 个专栏收录

26 篇文章

订阅专栏

深度学习

16 篇文章

订阅专栏

本文介绍在使用Keras框架保存包含Lambda层的模型时遇到的问题及解决方法。当使用ModelCheckpoint保存整个模型而非仅权重时，会遇到与Lambda层的冲突。解决策略是将save_weights_only设置为True。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在使用Keras框架，保存模型的时候，会遇到这样的错误：

原先保存模型的代码是：

model_path = "./ResNet/" + args.model + '_{epoch:03d}.hdf5'
checkpoints = ModelCheckpoint(model_path, monitor='val_loss', verbose=1, 
                              save_best_only=False,
                              save_weights_only=False, mode='auto', period=0)

报错原因是：网络结构中使用有Lambda层，这与ModelCheckpoint()的save_weights_only存在有冲突。

解决方法：将save_weights_only=False更换为save_weights_only=True

model_path = "./ResNet/" + args.model + '_{epoch:03d}.hdf5'
checkpoints = ModelCheckpoint(model_path, monitor='val_loss', verbose=1, 
                              save_best_only=False,
                              save_weights_only=True, mode='auto', period=0)