Rickzza的博客

原创 Pytorch实现mnist数据集图像自编码降噪（新人纯小白）

这篇文章出现的原因是：作为小白，在刚开始学习的时候，框架用的是Tensorflow，但由于一些原因，老师让将框架改为Pytorch，练习mnist数据集图像自编码降噪。而作者编程水平又是十分稀碎，让我自己从无到有编一个自编码器，可以说是不可能，无奈只能上CSDN求助大佬，希望能白嫖写好的代码。如果train_loss的训练值卡在某个数不动了，建议将学习率，也就是 lr = 0.001 改为0.0001，如果还不行，就改优化器。新人第一次发文，如果有不对的地方，还请各位大佬多多包涵！好了，不多说，上代码。