Skip to content

Comments

Cleaner workaround for max pool#103

Merged
drnikolaev merged 1 commit intoNVIDIA:caffe-0.14from
thatguymike:maxpool_workaround
Jan 20, 2016
Merged

Cleaner workaround for max pool#103
drnikolaev merged 1 commit intoNVIDIA:caffe-0.14from
thatguymike:maxpool_workaround

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