tf.saved_model.save报错解决

前几天写了一篇同样关于 tf.saved_model.save报错的解决文章,不过这次的错误不太一样。

上一篇文章:

savemodel报错

值得注意的是:

我使用的模型是自定义模型,而不是使用keras.model中的Sequential()封装的函数,所以不能使用model.save(‘xxx.h5’)对模型进行保存,只能通过tf.saved_model或者是save_weights来保存。

这次我同样是搭建GAN网络来破解口令,不同的是生成器方面,不再使用卷积核激活,而是使用LSTM网络代替,训练后保存模型时会报错:
TypeError: can’t pickle _thread.lock objects
在这里插入图片描述

这个错误搜索后发现是线程的问题,但是网上很难找到和模型保存相关的解决方案,于是我仔细查看了一下,发现只是生成器保存会出现问题,判别没有问题,再看一下错误提示,发现从deepcopy处产生的错误
在这里插入图片描述
查看模型代码后,我发现在自定义模型的call函数中,我为了区分两个变量,使用了copy.deepcopy函数,经过测试后我发现,只要在自定义模型的call函数中使用deepcopy函数,那么tf.saved_model.save就会报上面的错误,替换上述语句,这里我采用的是列表存储,之后代码就可以正常了。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值