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

Cleaner workaround for max pool #103

Merged
merged 1 commit into from
Jan 20, 2016

Conversation

thatguymike
Copy link

Only switch pooling engine on maxpool. Other cases should be okay. The source of the issue is how indexes are tracked to deal with handling multiple maxes (e.g. ties). Caffe makes cuDNN do this in place which can corrupt the indecies leading to arbitrary tie breaking. Until we can review, switch default engine in the case of max pool.

drnikolaev added a commit that referenced this pull request Jan 20, 2016
@drnikolaev drnikolaev merged commit 4d58e6d into NVIDIA:caffe-0.14 Jan 20, 2016
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

Successfully merging this pull request may close these issues.

2 participants