今天使用shuffleNetV2+，使用自己的数据集，遇到了loss是nan的情况，而且top1精确率出现断崖式上升，这显示是不正常的。
在网上查了下解决方案。我的问题是出在学习率上了。
我自己做的样本数据集比较小，就三类，每类大概三百多张，初始学习率是0.5。后来设置为0.1就解决了。
按照解决方案上写的。出现nan的情况还有以下几种：
学习率太大，但是样本数据集又很小。（我的情况）
自定义的loss除以了一个很小的数字，小到接近0。
数据不干净，数据本身就有nan，可以用numpy.isnan检查。
target，即label是大于等于0的。从1到类别数目-1变化。
以上这篇Pytorch训练过程出现nan的解决方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。
复制代码 代码如下: #!/usr/bin/env python import os, sys, time while True: time.sleep(3) try: ret = os...
python怎么查看数据类型? 第一步我们首先需要知道在python中查看变量数据类型是使用type()函数，type函数的用法如下图所示： 第二步我们打开py文件，输入...
re.sub 功能，比replace强大的替换函数，将正则表达式匹配上的模块替换成repl re.sub(pattern, repl, string, count=0, flags=0)...
python安装教程，分享给大家。 一、安装python 1、首先进入网站下载：点击打开链接（或自己输入网址），进入之后如下图，选择图中红色圈中区域进行下载。 2、下载完成后如下图所示...
如何实现自定义一个异常 python内置了许多异常类，为编写代码划定红线，才使调试代码时能及时发现错误。那么我们编写一个模块也可以为使用此模块者划定红线，来约束使用者可用哪些数据，这就需...