deeplearning4j-examples
deeplearning4j-examples copied to clipboard
Bug in ImportMobileNetExample
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.