seekFire commented on issue #19499: URL: https://github.com/apache/incubator-mxnet/issues/19499#issuecomment-725361524
@szha Well, I think this question is not important for me, as long as the model can be trained with at least one image per batch...BTW, when I run some script like below under the shell environment, it works OK: ``` >>> import mxnet as mx >>> x = mx.nd.ones((2, 3, 4, 5)) >>> y = x.transpose((0, 3, 1, 2)).reshape(0, -3, -1) >>> y.shape # (2, 15, 4) ``` But the reshape operation in my custom metric function will generate error during training: `labels = labels.transpose((0, 2, 3, 1)).reshape(0, -3, -1).argmax(-1)` The error message shown as below: `ValueError: can only specify one unknown dimension` So what do you think about the cause? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
