今天使用shuffleNetV2+，使用自己的数据集，遇到了loss是nan的情况，而且top1精确率出现断崖式上升，这显示是不正常的。
在网上查了下解决方案。我的问题是出在学习率上了。
我自己做的样本数据集比较小，就三类，每类大概三百多张，初始学习率是0.5。后来设置为0.1就解决了。
按照解决方案上写的。出现nan的情况还有以下几种：
学习率太大，但是样本数据集又很小。（我的情况）
自定义的loss除以了一个很小的数字，小到接近0。
数据不干净，数据本身就有nan，可以用numpy.isnan检查。
target，即label是大于等于0的。从1到类别数目-1变化。
以上这篇Pytorch训练过程出现nan的解决方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。
Python矩阵的基本用法 mat（）函数将目标数据的类型转化成矩阵（matrix） 1，mat()函数和array()函数的区别 Numpy函数库中存在两种不同的数据类型（矩阵ma...
本文实例讲述了python生成IP段的方法。分享给大家供大家参考。具体实现方法如下： #!/usr/local/bin/python #-*- coding: UTF-8 -*- #...
如果内部修改外部变量需要nonlocal，global def f1(): print("in f1..") num=111 def f2(): nonlocal num...
用函数包装来处理复杂的数据过滤 另一个常见的需求是按URL里的关键字来过滤数据对象。 之前，我们在URLconf中硬编码了出版商的名字，但是如果我们想用一个视图就显示某个任意指定的出版商...
首先来看一下代码： chars = "abcd" tmp = [] for char in chars: tmp.append(ord(char)) print(tmp) 这是一...