
pytorch
SweetBonk
一个人工智能小白
展开
-
训练准确率很高,预测准确率确很低的问题bug
最近接到一个二分类的任务,使用VGG训练的模型,结果发现训练准确率很高测试准确率去很低。不管怎么改权重改模型都一样。经过长时间的分析确定了训练的代码没问题,权重也没问题,那么结果就出在了预测代码上。最终发现是权重没有加载,等于我每次都用初始化的模型来预测的。大家遇到这种问题不妨看看权重有没有加载上net.load_state_dict(model)...原创 2022-07-12 15:38:09 · 2830 阅读 · 1 评论 -
2021-06-08
使用Auto Enconders 实现MNIST数据的生成自编码器是生成式深度学习的经典案例网络架构代码实现我的研究方向是医学图像分割,有一个想法是使用生成式的网络来对数据集进行预处理,使可供训练的图像增多,于是学习了自编码器,以后会更新AVE,GAN,CGAN等自编码器是生成式深度学习的经典案例自编码器的原理是通过对输入的x进行编码得到一个低维向量z,然后根据这个z通过解码器还原出x1。通俗来说是通过x得到x1。(看上去没啥用处,不妨想象一下通过x得出一个跟x非常相似又无关的x1,一个数据是否能衍生原创 2021-06-08 17:18:39 · 133 阅读 · 0 评论