fix weights rank assertion in InnerProductOp

pestisid-cmd · Feb 22, 2020 · c23ab37 · c23ab37
1 parent 96b26dc
commit c23ab37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/dnn/src/cuda4dnn/primitives/inner_product.hpp b/modules/dnn/src/cuda4dnn/primitives/inner_product.hpp
@@ -31,7 +31,7 @@ namespace cv { namespace dnn { namespace cuda4dnn {
             : stream(std::move(stream_)), cublasHandle(std::move(handle)), axis{ axis }
         {
             weightsTensor = csl::makeTensorHeader<T>(weights);
-            CV_Assert(get_effective_rank(weightsTensor) == 2);
+            CV_Assert(get_effective_rank(weightsTensor) <= 2);
             csl::copyMatToTensor<T>(weights, weightsTensor, stream);
 
             if (!bias.empty())