今天使用shuffleNetV2+，使用自己的数据集，遇到了loss是nan的情况，而且top1精确率出现断崖式上升，这显示是不正常的。
在网上查了下解决方案。我的问题是出在学习率上了。
我自己做的样本数据集比较小，就三类，每类大概三百多张，初始学习率是0.5。后来设置为0.1就解决了。
按照解决方案上写的。出现nan的情况还有以下几种：
学习率太大，但是样本数据集又很小。（我的情况）
自定义的loss除以了一个很小的数字，小到接近0。
数据不干净，数据本身就有nan，可以用numpy.isnan检查。
target，即label是大于等于0的。从1到类别数目-1变化。
以上这篇Pytorch训练过程出现nan的解决方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。
set集合 无序可变 由不同元素组成 其元素必须为可哈希的类型（通俗来说不可变类型） 集合的两种定义方式 使用{} Eg：{1,2,3,4,5} 使用set(可...
场景：集团中心下发本省数据时，并未按地市、业务拆分，现需要按地市、业务拆分并分发到地市。 本文利用Python的pandas包实现了以上场景。 注：本示例代码只实现按单列拆分，如果需要多...
PyTorch基础入门三：PyTorch搭建多项式回归模型 1）理论简介 对于一般的线性回归模型，由于该函数拟合出来的是一条直线，所以精度欠佳，我们可以考虑多项式回归来拟合更...
MASK图像掩膜处理 在图像操作中有时候会用到掩膜处理，如果使用遍历法掩膜图像ROI区域对于python来讲是很慢的，所以我们要找到一种比较好的算法来实现掩膜处理。 假设我们有一副图...
python中eval和int的区别是什么？下面给大家介绍一下: 1.eval()函数 eval(<字符串>)能够以Python表达式的方式解析并执行字符串，并将返回结果输出...