关于“训练时效果还可以,但测试模型时fid很高,效果很差”的分析解决

同前文,我是处于改模型的阶段,参考之前的笔记保存模型的时候我把model_dict[name] = net.module.cpu().state_dict()改为了model_dict[name] = net.cpu().state_dict(),就是把module这个删掉了。也不知道这个module是什么意思,总之我个人分析,改了以后的应该是把模型保存到cpu中了,反正没有module这个东西。
跑代码的过程中,我发现效果是可以的(不管和原论文比有没有提升,但绝对不是一堆噪声),但在测试的时候我发现生成了一堆噪声,且FID分数为三四百,相当于翻译的图像和源图像几乎无关(风格迁移),所以肯定是加载模型出问题了,加载的不是我保存的模型。
原因:在这里插入图片描述红框里是我注释掉的,注释后就可以正常测试了。因为执行了if语句里的东西,所以加载模型是还是net.module.load_state_dict(),但保存时我没有使用module。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值