-
Notifications
You must be signed in to change notification settings - Fork 10
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
ONNX loader doesn't load the BatchNorm parameters epsilon
and momentum
?
#305
Comments
They are almost the same, there is a certain loss of precision. Reproducing the issue: And they must be loaded because the default values in ONNX are momentum=0.9 and epsilon=1e-5.
|
Right, I was loading the ONNX incorrectly. |
Yes, you are right. I am going to change it. |
I checked the ONNX operators definition and it is correct now, I don't have to change it. You can see here that the ONNX definiton uses the momentum like the EDDL. This means that if you use a momentum of 0.1 in Pytorch, It should export it as momenum=0.9. But the EDDL is importing it right. |
To Reproduce
Steps to reproduce the behavior:
layers[2]
Expected behavior
They should be the same.
The text was updated successfully, but these errors were encountered: