-
Notifications
You must be signed in to change notification settings - Fork 18.7k
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
im2col with hole #3232
im2col with hole #3232
Conversation
…fit on current caffe implementation.
👍 |
__CUDA_ARCH__ is not defined in host code; the #if was vacuous and misleading.
Remove dead preprocessor code for number of CUDA threads
[build] Display and store cuDNN version numbers for CMake
don't divide by 0 duration when downloading model binary
Remove hamming_distance and popcount
@tamakoji Would you please rebase this PR with the current caffe? |
…fit on current caffe implementation.
Conflicts: include/caffe/vision_layers.hpp
@llllaaaa done. |
@tamakoji Thanks a lot! @shelhamer Could you merge this into the master branch? I think it is a very useful feature |
there are lots of lint errors(mostly exceeding 80chars), do I need to fix them? |
@tamakoji Thank you for your works, but the code cannot run in the cudnn mode. What should I do to make the code support cudnn? |
@GoYchen cudnn does not use im2col, and does not support holes |
@llllaaaa Thank you very much for your answer. |
Ported algorithm of George Papandreou(DeepLab)' im2col with hole to fit on current caffe
implementation. variable name "hole" been modified to 'kernel_stride" which make more sense than
hole, since hole = 1 does not have hole, its just kernel stride.
modified test_im2col_kernel.cu and verified(tested test_im2col_kernel and test_im2col_layer).