GAN的loss为nan的解决办法(成功)

跑了好几个GAN模型,一运行loss就巨大,几次epoch后就变NaN,或者直接就爆掉

有人说loss为nan是因为里面有log出问题了,但我想着文章里给的代码肯定得是能跑的啊

后来换了个电脑,重新把cyclegan的源码跑了一下,又完全好了

震惊!

后来我对比了一下底层,发现能跑通的这个跟其他的连conv2d的底层都不一样,所以判断是package的版本不一样

于是我把所有其他GAN代码的环境都变成能跑通的这个

其他的GAN也就都能跑通了

相关推荐
Tutankaaa3 小时前
从被动接受到主动挑战:知识竞赛如何重塑学习价值
人工智能·经验分享·笔记·学习
Jmayday4 小时前
机器学习基本理论
人工智能·机器学习
ZhengEnCi4 小时前
01b-上下文向量与信息瓶颈
人工智能
王_teacher4 小时前
机器学习 矩阵求导 完整公式+严谨推导
人工智能·线性代数·考研·机器学习·矩阵·线性回归
码以致用4 小时前
DeerFlow Memory架构
人工智能·ai·架构·agent
ting94520004 小时前
从零构建大模型实战:数据处理与 GPT-2 完整实现
人工智能
学点程序4 小时前
Manifest:帮个人 AI Agent 降低模型成本的开源路由器
人工智能·开源
可观测性用观测云5 小时前
观测云 x AI Agent:运维智能化的范式跃迁实践
人工智能
数数科技的数据干货5 小时前
ThinkingAI携手华为云,共建企业级AI Agent平台Agentic Engine
人工智能·ai·华为云·agent
人工智能AI技术5 小时前
春招急救:7天面试突击方案
人工智能