diff --git a/loadcaffe.cpp b/loadcaffe.cpp index 500202d..bd8f8ad 100644 --- a/loadcaffe.cpp +++ b/loadcaffe.cpp @@ -561,7 +561,6 @@ void convertProtoToLuaV2(const caffe::NetParameter &netparam, const char* lua_na lines.emplace_back(layer.name(), "cudnn.SoftMax()"); else lines.emplace_back(layer.name(), "nn.SoftMax()"); - break; } if(layer.type()=="Softmax") { @@ -569,7 +568,6 @@ void convertProtoToLuaV2(const caffe::NetParameter &netparam, const char* lua_na lines.emplace_back(layer.name(), "cudnn.SoftMax()"); else lines.emplace_back(layer.name(), "nn.SoftMax()"); - break; } if(!lines.empty())