v1版本训练梯度爆炸

Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper

MIT License

3 stars 0 forks source link

v1版本训练梯度爆炸 #80

Open nobody-cheng opened 5 years ago

nobody-cheng commented 5 years ago

您好，请问我自己的数据用在无anchorsv2版本上能正常跑通，但在v1版本刚开始训练就抛出异常梯度爆炸，是由于什么原因导致 print "running" if np.isnan(loss_val): raise ArithmeticError('The gradient is exploded')

123lifei commented 5 years ago

@nobody-cheng 问题找到了吗？我也是这种情况

nobody-cheng commented 5 years ago

@123lifei 暂时没有，我训练37042776尺寸的图片没有问题，现在就是快3w步 loss 不收敛，维持在200上下，使用30722048尺寸的数据集会出现nan情况，还在找原因