Fix gradient accumulation for cuDNN convolution layer#50
Fix gradient accumulation for cuDNN convolution layer#50lukeyeager merged 1 commit intoNVIDIA:masterfrom
Conversation
|
Why not just cherry-pick @ronghanghu's commit? Commenting rather than deleting those lines may lead to a merge conflict down the road. Also, why did you change the test if it wasn't needed in BVLC/pull/3254? |
|
I changed the gradient file to match what is in bvlc today, and without it the convolution tests fail
|
|
Oh sorry, I didn't notice this was for the So we need this fix in the |
Fix gradient accumulation for cuDNN convolution layer
|
Merged into |
Fixes the gradient accumulation issue in CuDNNConvolutionLayer reported in caffe BVLC/pull/3254