Shape mismatch in Java API

Gluon
1

Hi, I’ve finished training the yolo3_darknet53 params and wanted to test it with Java API, however am having this error which I have no idea where it coming from:
Exception in thread “main” java.lang.IllegalArgumentException: requirement failed:

    Mismatch between     shape (100,1) and (1,100,4)
        	at scala.Predef$.require(Predef.scala:224)
        	at org.apache.mxnet.NDArray$$anonfun$9.apply(NDArray.scala:448)
        	at org.apache.mxnet.NDArray$$anonfun$9.apply(NDArray.scala:445)
        	at scala.collection.TraversableLike$$anonfun$map$1.apply(TraversableLike.scala:234)
        	at scala.collection.TraversableLike$$anonfun$map$1.apply(TraversableLike.scala:234)
        	at scala.collection.Iterator$class.foreach(Iterator.scala:893)
        	at scala.collection.AbstractIterator.foreach(Iterator.scala:1336)
        	at scala.collection.IterableLike$class.foreach(IterableLike.scala:72)
        	at scala.collection.AbstractIterable.foreach(Iterable.scala:54)
        	at scala.collection.TraversableLike$class.map(TraversableLike.scala:234)
        	at scala.collection.AbstractTraversable.map(Traversable.scala:104)
        	at org.apache.mxnet.NDArray$.concatenate(NDArray.scala:445)
        	at org.apache.mxnet.module.BaseModule$$anonfun$2.apply(BaseModule.scala:267)
        	at org.apache.mxnet.module.BaseModule$$anonfun$2.apply(BaseModule.scala:267)
        	at scala.collection.TraversableLike$$anonfun$map$1.apply(TraversableLike.scala:234)
        	at scala.collection.TraversableLike$$anonfun$map$1.apply(TraversableLike.scala:234)
        	at scala.collection.mutable.ResizableArray$class.foreach(ResizableArray.scala:59)
        	at scala.collection.mutable.ArrayBuffer.foreach(ArrayBuffer.scala:48)
        	at scala.collection.TraversableLike$class.map(TraversableLike.scala:234)
        	at scala.collection.AbstractTraversable.map(Traversable.scala:104)
        	at org.apache.mxnet.module.BaseModule.predict(BaseModule.scala:267)
        	at org.apache.mxnet.infer.Predictor$$anonfun$11.apply(Predictor.scala:210)
        	at org.apache.mxnet.infer.Predictor$$anonfun$11.apply(Predictor.scala:210)
        	at org.apache.mxnet.infer.MXNetThreadPoolHandler$$anon$4.call(MXNetHandler.scala:73)
        	at java.util.concurrent.FutureTask.run(FutureTask.java:266)
        	at java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1149)
        	at java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:624)
        	at java.lang.Thread.run(Thread.java:748)

    FAILURE: Build failed with an exception.

    * What went wrong:
    Execution failed for task ':ObjectDetectionTutorial.main()'.
    > Process 'command '/usr/lib/jvm/java-8-openjdk-amd64/bin/java'' finished with non-zero exit value 1

    * Try:
    Run with --stacktrace option to get the stack trace. Run with --info or --debug option to get more log output. Run with --scan to get full insights.

    * Get more help at https://help.gradle.org

BUILD FAILED in 1s
2 actionable tasks: 2 executed
Process 'command '/usr/lib/jvm/java-8-openjdk-amd64/bin/java'' finished with non-zero exit value 1
10:16:24: Task execution finished 'ObjectDetectionTutorial.main()'.

I only have 1 class and my input image is 512 and pretrained_base=false. Any help would be much appreciated. Thank you in advance.