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

Document and Test Convolution #1104

Merged
merged 3 commits into from
Sep 18, 2014
Merged

Conversation

shelhamer
Copy link
Member

To thoroughly check convolution, the output is compared against a reference implementation by explicit looping. Simple and group convolution by the Caffe and cuDNN engines are checked against the reference.

The reference convolution rolls 8 loops deep.

  • comment ConvolutionLayer / im2col
  • comment CuDNNConvolutionLayer
  • switch to random weights instead of averaging

@shelhamer
Copy link
Member Author

This is done Switching to random weights for more robust checking.

* inputs so that the im2col matrix has a column for each input region to
* be filtered. col2im restores the output spatial structure by rolling up
* the output channel N' columns of the output matrix.
*
* TODO(dox): thorough documentation for Forward, Backward, and proto params.
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

rm TODO, nice docs!

@jeffdonahue
Copy link
Contributor

probably worth switching to random weights, but I'll leave it up to you. otherwise LGTM!

To thoroughly check convolution, the output is compared against a
reference implementation by explicit looping. Simple and group
convolution by the Caffe and cuDNN engines are checked against the
reference.
@shelhamer
Copy link
Member Author

Switched to random weights and uncovered a grouping issue in my reference convolution, so yeah, it was worth it.

Ready for merge once Travis oks it.

shelhamer added a commit that referenced this pull request Sep 18, 2014
@shelhamer shelhamer merged commit 8dac339 into BVLC:dev Sep 18, 2014
@shelhamer shelhamer deleted the conv-comments-tests branch September 18, 2014 16:45
mitmul pushed a commit to mitmul/caffe that referenced this pull request Sep 30, 2014
RazvanRanca pushed a commit to RazvanRanca/caffe that referenced this pull request Nov 4, 2014
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants