fix weights rank assertion in InnerProductOp

5 years ago · c23ab37355
parent 96b26dc8f4
commit c23ab37355
1 changed files with 1 additions and 1 deletions
--- 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())