reminisce opened a new pull request #7682: Fix shape inference bug URL: https://github.com/apache/incubator-mxnet/pull/7682 This PR fixes a shape inference bug in function `InferAttr`. Consider the following network: ``` weight --> abs_op --> sum_op \ add_op / data --> fc_op --> sum_op ``` Given a data shape, if the shape forward inference starts from weight, there would be several node entries with unknown shapes for abs_op and sum_op. The weight shape is only known after fc_op's shape inference is done. When the backward shape inference starts, it assumes that all forward node entries' shapes are known. However, this is not true in this case and therefore, leads to `CHECK_EQ` failure. The problem is fixed by skipping `CHECK_EQ` for empty shapes in the backward shape inference. @piiswrong @eric-haibin-lin ---------------------------------------------------------------- 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