Skip to content
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.

Sign up for GitHub

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

Jump to bottom

Is there support for BatchNorm? #51

Closed
anirudth opened this issue May 21, 2020 · 2 comments
Closed

Is there support for BatchNorm? #51

anirudth opened this issue May 21, 2020 · 2 comments
Assignees
@quic-sundarr
Labels
question Further information is requested

Comments

@anirudth
Copy link

I get the following error 2020-05-21 16:16:35,240 - Winnow - ERROR - Unsupported op_type Mul, dotted net/Conv2d/BatchNorm/batchnorm_1/mul_1, input_ops: net/Conv2d/Conv2D]
@quic-ssiddego quic-ssiddego added the question Further information is requested label May 21, 2020
@quic-klhsieh
Copy link
Contributor

Hi @anirudth, if your model is using non-fused BatchNormalization ops, support for this was added earlier today. Can you please retry with the latest code and see if you still encounter this issue?

Please note that we also only support BatchNorm ops created using tf slim or tf Keras apis. Can you provide any additional details about the model you are using?

@anirudth
Copy link
Author

Thanks for the quick reply. It seems to work.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@quic-sundarr quic-sundarr

Labels
question Further information is requested
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
4 participants
@anirudth @quic-klhsieh @quic-ssiddego @quic-sundarr