-
Notifications
You must be signed in to change notification settings - Fork 18.6k
Pooling factory #1172
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
Pooling factory #1172
Conversation
Looks good, except shall we preserve the fallback warning at layer creation time? (I guess if the engine is specified as |
Done. I made it LOG(INFO) since it is not a crucial issue. |
@sguada Sergio if this looks good to you I'll merge and close the other PR. |
@Yangqing looks good to me |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This does not compile for me,
This does:
CHECK_EQ(this->pad_h_, 0);
Now this commit fails tests: |
After BVLC#1172, some cuDNN pooling layer tests were failing due to use of padding. (Before BVLC#1172, these tests were actually testing PoolingLayer instead of CuDNNPoolingLayer via the fallback.) This commit disables many tests via commenting, so that they can be easily readded when cuDNN gains pooling padding support.
Pooling factory
After BVLC#1172, some cuDNN pooling layer tests were failing due to use of padding. (Before BVLC#1172, these tests were actually testing PoolingLayer instead of CuDNNPoolingLayer via the fallback.) This commit disables many tests via commenting, so that they can be easily readded when cuDNN gains pooling padding support.
Pooling factory
After BVLC#1172, some cuDNN pooling layer tests were failing due to use of padding. (Before BVLC#1172, these tests were actually testing PoolingLayer instead of CuDNNPoolingLayer via the fallback.) This commit disables many tests via commenting, so that they can be easily readded when cuDNN gains pooling padding support.
After BVLC#1172, some cuDNN pooling layer tests were failing due to use of padding. (Before BVLC#1172, these tests were actually testing PoolingLayer instead of CuDNNPoolingLayer via the fallback.) This commit disables many tests via commenting, so that they can be easily readded when cuDNN gains pooling padding support.
After BVLC#1172, some cuDNN pooling layer tests were failing due to use of padding. (Before BVLC#1172, these tests were actually testing PoolingLayer instead of CuDNNPoolingLayer via the fallback.) This commit disables many tests via commenting, so that they can be easily readded when cuDNN gains pooling padding support.
This is a PR to address the comments at #1170.