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.

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

BatchNorm consuming a lot of time #66

Open
leochli opened this issue Jun 11, 2018 · 2 comments
Open

BatchNorm consuming a lot of time #66

leochli opened this issue Jun 11, 2018 · 2 comments

Comments

@leochli
Copy link

leochli commented Jun 11, 2018

Hi shicai,

Have you evaluated the time performance of mobilenetv1 & v2?
I found that the bn layer and linear layer are very time consuming both in CPU-only and GPU-only cases. Is the eps setting really necessary for the BN layer?

@basaltzhang
Copy link

Try to use the DepthwiseConvolution layer instead of Convolution type, if the group parameter is not 1. The DepthwiseConvolution implementation can be found here

@AnPay
Copy link

AnPay commented May 2, 2019

Can someone update the time performance for mobilenets?
i am getting it ~300ms for googlenet. Want to verify if its correct. tested on S10

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants