piiswrong commented on issue #9842: Custom Function Shape Inference URL: https://github.com/apache/incubator-mxnet/issues/9842#issuecomment-369063273 Actually. This is not a MXNet bug. Although the error message is not clear. What happened is Conv2D block relies on the mx.sym.Convolution operator to figure out weight shape from data. Adding a custom op to weight blocks that shape inference path. You can solve this by specifying the in_channels argument for Conv2D. We should improve the error message and report "Deferred initialization failed because xx's shape cannot be inferred" @sxjscience
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services