yjxiong / caffe

A fork of Caffe with OpenMPI-based Multi-GPU (mainly data parallel) support for action recognition and more. More documentation please see the original readme.
http://caffe.berkeleyvision.org/
Other
551 stars 153 forks source link

fix the bug in input reshape of cuDNN #172

Closed yjxiong closed 7 years ago

yjxiong commented 7 years ago

The previous cuDNN update created a bug where the net will produce incorrect results after input reshaping. This PR fixes this bug.