Bug in ImportMobileNetExample
MrForum opened this issue · 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.