diff --git a/src/relay/op/nn/convolution.h b/src/relay/op/nn/convolution.h index 52958c2824b2..5c67a047487f 100644 --- a/src/relay/op/nn/convolution.h +++ b/src/relay/op/nn/convolution.h @@ -1107,7 +1107,8 @@ bool Conv2DTransposeRel(const Array& 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(wshape); + << " out_channels=" << param->channels << " groups=" << param->groups + << " weight.shape=" << Array(wshape); } if (!dshape_nchw[1].as() && !wshape[0].as()) { ICHECK(reporter->AssertEQ(dshape_nchw[1], wshape[0]))