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

Use cuDNN routine FindEx to find the best algorithm. #158

Merged
merged 1 commit into from
Jun 8, 2016
Merged

Use cuDNN routine FindEx to find the best algorithm. #158

merged 1 commit into from
Jun 8, 2016

Commits on Jun 8, 2016

  1. Use cuDNN routine FindEx to find best algorithm. 

    FindEx is more stable than Get (heuristic-based) because it runs all the available algorithms and sorts them according to their speed.
Both Get and FindEx are supported now and can be specified through the definition of each layer in prototxt.
In Reshape, check whether shape of bottom and convolution descriptors have changed.
In caffe time, do multiple (instead of one) fwd/bwd pass in the initilization phase.
This is crucial because FindEx is executed in the first iterations and it takes quite a long time.
    @pooyadavoodi
    pooyadavoodi committed Jun 8, 2016
    Configuration menu
    Copy the full SHA
    4248b1f View commit details
    Browse the repository at this point in the history