benqua 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-334054179 Yes, I am conducting a segmentation tasks. My label has only one channel, with value being either 1 or 0, depending on the class (providedLabel: softmax_label -> (1,1,868,868)). The output of outputShapes on the module is OutShapes (module): ArrayBuffer((softmaxoutput0_output,(1,2,868,868))) You mean I shouldn't use the multi_output parameter? ---------------------------------------------------------------- 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