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

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

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

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

震惊!

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

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

其他的GAN也就都能跑通了

相关推荐
武子康几秒前
MiniMax H3 的多模态参考为什么比 2K 更重要?
人工智能·llm·agent
IT爱学堂2 分钟前
AI Agent大师之路:从认知架构到自主智能体的完整设计哲学
人工智能
想会飞的蒲公英7 分钟前
PyTorch中SGD 与 Momentum 从零理解:给最朴素的优化器加上“惯性“
人工智能·pytorch·python·深度学习·机器学习
额恩668 分钟前
自然语言处理 NLP 入门与语言学基础
人工智能·自然语言处理
paopaokaka_luck9 分钟前
基于springboot3+vue3的乡村医生诊疗管理系统(AI助手、协同过滤算法、webSocket实时聊天、Echarts图形化分析)
前端·网络·人工智能·spring boot·websocket·网络协议·echarts
用户2986985301410 分钟前
PDF 转图片?三种方案,覆盖全平台与自动化场景
人工智能·后端
Coffeeee11 分钟前
天天 AI Coding 的你,如果出去面试,你的竞争力是什么?
人工智能·程序员·ai编程
小飞猪。。18 分钟前
笔记十八:大模型 RLHF 系统工程实战笔记
人工智能·笔记
钱六两19 分钟前
Spring AI 使用 MCP 客户端(调用高德 MCP)
java·人工智能·spring
盈飞无限19 分钟前
AI智能SPC重构制程管控逻辑,打造质量硬核底座
大数据·人工智能·重构