search

faustomilletari / VNet

GNU General Public License v3.0
284 stars 123 forks source link

Some errors I meet in the training #57

Open 1348205 opened 6 years ago

1348205 commented 6 years ago

Check failed: 2 == this->num_spatialaxes (2 vs. 3) CuDNNConvolution input must have 2 spatial axes (e.g., height and width). Use 'engine: CAFFE' for general ND convolution. Check failure stack trace:

Hi, I check several possibilities for the error, and also try use only cpu, but the error is still exist. Can you some suggestions for this error, Thank you very much

1348205 commented 6 years ago

cudnn_conv_layer.cpp:94] Check failed: 2 == this->num_spatialaxes (2 vs. 3) CuDNNConvolution input must have 2 spatial axes (e.g., height and width). Use 'engine: CAFFE' for general ND convolution. Check failure stack trace:

This is the whole error statement

HainKUL commented 5 years ago

Hi, I also encountered the same problem. did you solve it finally?