问题截图
问题分析
分析:这个bug有几个奇怪的地方:
1、网络在训练的时候,出现了loss = inf 的情况,是在epoch = 1 的中间部分,出现了loss = inf 的情况,接下来的每一轮都是loss = inf的情况;
2、网络训练到一半突然报错;
3、报错的问题,根据网上的解决方法并没有解决。
Debug的思路:
1、之前网络复现的时候,是可以训练结束的,同时得到相应的结果。 但是我也发现了一些问题,就是loss也是inf,但是可以得到相应的结果。 因为是复现,所以我对结果并没有多大的概念,后来熟悉了原理之后,发现其实这时候就存在问题了,就是结果偏差。
2、因为网络是可以跑得通的,所以问题只能出现在数据集上面。后来通过学长指导发现,我把数据为0的图片也放入了网络中训练,最终导致了这个结果。
解决办法:
将数据集中为0的图片层清除掉,清除掉之后再放入网络中训练,即可解决这个问题!同时,得到的结果也会相应的变好!
未剔除0层:
剔除0层: