deeplearning4j-examples icon indicating copy to clipboard operation
deeplearning4j-examples copied to clipboard

Bug in ImportMobileNetExample

Open MrForum opened this issue 3 years ago • 0 comments

I get the following Exception running ImportMobileNetExample with a native backend:

Exception in thread "main" java.lang.IllegalArgumentException: Incorrect number of arguments for permute function: got arguments [0, 2, 3, 1] for rank 5 array. Number of arguments must equal array rank at org.nd4j.common.base.Preconditions.throwEx(Preconditions.java:636) at org.nd4j.common.base.Preconditions.checkArgument(Preconditions.java:137) at org.nd4j.linalg.api.ndarray.BaseNDArray.permute(BaseNDArray.java:4758) at mobilenet.ImportMobileNetExample.inceptionPreprocessing(ImportMobileNetExample.java:104) at mobilenet.ImportMobileNetExample.main(ImportMobileNetExample.java:38)

Code throwing the exception: // change to channels-last img = img.permute(0, 2, 3, 1);

Any help would be appreciated.

MrForum avatar May 03 '21 21:05 MrForum