We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5533af3 commit add6984Copy full SHA for add6984
src/caffe/layer_factory.cpp
@@ -165,7 +165,8 @@ shared_ptr<Layer<Dtype> > GetSqueezeConvolutionLayer(
165
return shared_ptr<Layer<Dtype> >(new SqueezeConvolutionLayer<Dtype>(param));
166
#ifdef USE_CUDNN
167
} else if (engine == SqueezeConvolutionParameter_Engine_CUDNN) {
168
- return shared_ptr<Layer<Dtype> >(new CuDNNSqueezeConvolutionLayer<Dtype>(param));
+ //return shared_ptr<Layer<Dtype> >(new CuDNNSqueezeConvolutionLayer<Dtype>(param));
169
+ return shared_ptr<Layer<Dtype> >(new SqueezeConvolutionLayer<Dtype>(param));
170
#endif
171
} else {
172
LOG(FATAL) << "Layer " << param.name() << " has unknown engine.";
0 commit comments