search

NVIDIA / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
672 stars 263 forks source link

BVLC model can directly used? #511

Closed HaoLiuHust closed 6 years ago

HaoLiuHust commented 6 years ago

I am using a bvlc model, but when init, the last layer have an error: Check failed: axis_index < num_axes() (2 vs. 2) axis 2 out of range for 2-D Blob with shape 1 512 (512)

HaoLiuHust commented 6 years ago

solved by modify cudnn_batch_norm_layer.cpp