Skip to content

Commit add6984

Browse files
committed
MCW patch fix: remove the usage of unimplemented CuDNNSqueezeConvolutionLayer.
1 parent 5533af3 commit add6984

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/caffe/layer_factory.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,8 @@ shared_ptr<Layer<Dtype> > GetSqueezeConvolutionLayer(
165165
return shared_ptr<Layer<Dtype> >(new SqueezeConvolutionLayer<Dtype>(param));
166166
#ifdef USE_CUDNN
167167
} else if (engine == SqueezeConvolutionParameter_Engine_CUDNN) {
168-
return shared_ptr<Layer<Dtype> >(new CuDNNSqueezeConvolutionLayer<Dtype>(param));
168+
//return shared_ptr<Layer<Dtype> >(new CuDNNSqueezeConvolutionLayer<Dtype>(param));
169+
return shared_ptr<Layer<Dtype> >(new SqueezeConvolutionLayer<Dtype>(param));
169170
#endif
170171
} else {
171172
LOG(FATAL) << "Layer " << param.name() << " has unknown engine.";

0 commit comments

Comments
 (0)