sobel-operator-pytorch icon indicating copy to clipboard operation
sobel-operator-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon chaddy1004

Add epsilon

Open MordehayM opened this issue 3 years ago • 0 comments

Hi, thanks for the code. you need to add to the sqrt, small number epsilon. Otherwise you will get in the backpropogation nan values. https://github.com/chaddy1004/sobel-operator-pytorch/blob/eca741aafbcef9d2639acedca10b09bbfa100e09/model.py#L20

MordehayM avatar Jul 24 '22 22:07 MordehayM