Home IT技术 我的PyTorch GAN从生成随机噪声变为生成全黑图像,且没有收敛。这是为什么? 我的PyTorch GAN从生成随机噪声变为生成全黑图像,且没有收敛。这是为什么? IT技术 xiaolong · 2025年5月30日 · 0 Comment 我的代码非常基础,只是按照DeepLearning.ai的GAN专项课程的第一个实验室进行的。然而,我的代码输出与预期不同,这是什么原因?如果这只是一个愚蠢的错误,我感到抱歉,这是我第一次接触GAN。我首先创建了生成器和判别器类,我的随机噪声函数,并创建了我的模型。然后我运行了训练循环,但在3个epoch后,GAN的所有输出都是黑色的。 ... 回答: 你的判别器损失计算有误。真实图像的标签应该是1而不是0。 更新后的代码: ... 对我来说,输出的图像看起来相当不错: 相关文章: 如何将一个模型的中间层传递给另一个模型以实现PyTorch中的跳跃连接 将PyTorch转换为ONNX JS时出现”无法对给定张量使用matmul”错误 如何在GAN中平衡生成器和判别器的性能? softmax和log-softmax有什么区别? 理解PyTorch CNN通道 PyTorch中的`CrossEntropyLoss()`函数 如何在PyTorch中初始化权重? PyTorch: Variable数据必须是张量 — 数据已是张量 如何在PyTorch中手动计算整个数据集的误差? KL Divergence for two probability distributions in PyTorch generative-adversarial-network machine-learning pytorch