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

model.predict(input_img) very slow #19

Open
mingmingDiii opened this issue Nov 4, 2017 · 2 comments
Open

model.predict(input_img) very slow #19

mingmingDiii opened this issue Nov 4, 2017 · 2 comments

Comments

@mingmingDiii
Copy link

@michalfaber Thanks for your great work!!
When I test the image 'ski.jpg'(shape: 7126743), 'model.predict(input_img)' took about 1200ms with a TITAN X GPU(only with scale 1). But in the caffe version, 'output_blobs = net.forward()' only took about 72ms. Can you help me figure it out? Thanks a lot!!

@yokings
Copy link

yokings commented Nov 7, 2017

you are right! I have meeted the same situation.Someone can help me figure it out?

@michalfaber
Copy link
Owner

@mingmingDiii @allenwangcheng I executed several iterations of model.predict(input_img) - only with scale 1 and shape 712x673x3. The first iteration took around 2300ms but all next iterations around 260ms (GPU 1070). I am not sure but there may be an additional cost of recompiling computation graph. I noticed that there is a latency peak every time when input image size changes (scale = multiplier[m]).

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