update logging information in convolution.h

apache · Nov 11, 2021 · 3e0db60 · 3e0db60
1 parent 562fdab
commit 3e0db60
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/relay/op/nn/convolution.h b/src/relay/op/nn/convolution.h
@@ -1107,7 +1107,8 @@ bool Conv2DTransposeRel(const Array<Type>& types, int num_inputs, const Attrs& a
       ICHECK(reporter->AssertEQ(indexdiv(param->channels, param->groups), wshape[1]))
           << "Conv2DTransposed: shape of weight is inconsistent with out_channels, "
           << " out_channels // groups != weight.shape[1] "
-          << " out_channels=" << param->channels << " weight.shape=" << Array<IndexExpr>(wshape);
+          << " out_channels=" << param->channels << " groups=" << param->groups
+          << " weight.shape=" << Array<IndexExpr>(wshape);
     }
     if (!dshape_nchw[1].as<tir::AnyNode>() && !wshape[0].as<tir::AnyNode>()) {
       ICHECK(reporter->AssertEQ(dshape_nchw[1], wshape[0]))