search

gangweiX / ACVNet

[TPAMI 2024 & CVPR 2022] Attention Concatenation Volume for Accurate and Efficient Stereo Matching
MIT License
458 stars 63 forks source link

关于梯度消失和梯度爆炸 #25

Closed zhao976 closed 1 year ago

zhao976 commented 2 years ago

作者您好，我在sceneflow数据集上训练ACVNet时，出现了masks[idx].float().mean() too small的警告，在冻结权值训练时出现了NaN or Inf found in input tensor的问题，判断为出现梯度爆炸，学习率设置为0.0001也没有用。请问这种情况如何解决？

mooncii commented 1 year ago

您好 我在自己的数据集上也出现too small的问题，请问您现在解决了吗