deeplearning4j / deeplearning4j-examples

Deeplearning4j Examples (DL4J, DL4J Spark, DataVec)

Home Page:http://deeplearning4j.konduit.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.