Ldpe2G commented on issue #8129: [scala] Module api: label (1,1,868,868) and prediction (1,868,868)should have the same length URL: https://github.com/apache/incubator-mxnet/issues/8129#issuecomment-334036151 @benqua are you conducting the segmentation task? If so, the shape of label has only one channel, just like the normal softmax. The multi_output parameter means to calculate the softmax along the channel axis . And I suggest you first print the output shapes by calling the `outputShapes` method of Module just after you call the `bing` method. ---------------------------------------------------------------- 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
