[python.onnx ] save model
%122 = take(%121, %v635, axis=0); %123 = expand_dims(%122, axis=0); %124 = shape_of(%blocks.1.0.weight, dtype="int64"); %125 = take(%124, %v647, axis=0); %126 = expand_dims(%125, axis=0); %127 = reshape(%120, newshape=[1, 4096, 4, 4]); %128 = reshape(%blocks.1.0.weight, newshape=[1, 512, 512, 3, 3]); %129 = multiply(%128, %v678); %130 = take(%93, %v364, axis=1); %131 = multiply(%blocks.1.0.fc.weight, %v655); %132 = transpose(%131, axes=[1, 0]); %133 = transpose(%132, axes=[1, 0]); %134 = nn.dense(%130, %133, units=None); %135 = reshape(%blocks.1.0.bias.bias, newshape=[1, 512]); %136 = add(%134, %135); %137 = add(%136, %v668); %138 = reshape(%137, newshape=[8, 512, 1, 1, 1]); %139 = multiply(%129, %138); %140 = power(%139, %v691); %141 = sum(%140, axis=[1, 3, 4]); %142 = add(%141, %v694); %143 = power(%142, %v696); %144 = divide(%v698, %143); %145 = reshape(%144, newshape=[8, 1, 512, 1, 1]); %146 = multiply(%139, %145); %147 = reshape(%146, newshape=[4096, 512, 3, 3]); %148 = nn.conv2d_transpose(%127, %147, channels=512, kernel_size=[3, 3], strides=[2, 2], padding=[0, 0, 0, 0], groups=8) in particular dimension 1 conflicts 64 does not match 512; unable to unify: `Tensor[(4096, 64, 3, 3), float32]` and `Tensor[(4096, 512, 3, 3), float32]`; ; %149 = shape_of(%148, dtype="int64"); %150 = take(%149, %v728, axis=0); %151 = expand_dims(%150, axis=0); %152 = shape_of(%148, dtype="int64"); %153 = take(%152, %v731, axis=0); %154 = expand_dims(%153, axis=0); %155 = (%123, %126, %151, %154); concatenate(%155) --- [Visit Topic](https://discuss.tvm.ai/t/pytorch-onnx-model-nn-conv2-in-particular-dimension-1-conflicts-4096-does-not-match-512/6255/2) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/a22315253e4bad3e0e1e61963a34f0459c8c50e6e766cb5a765e36afdbd6e3ac).
