You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@mxnet.apache.org by GitBox <gi...@apache.org> on 2018/12/05 01:54:54 UTC

[GitHub] andrewfayres commented on a change in pull request #13522: [MXNET-1249] Fix Object Detector Performance with GPU

andrewfayres commented on a change in pull request #13522: [MXNET-1249] Fix Object Detector Performance with GPU
URL: https://github.com/apache/incubator-mxnet/pull/13522#discussion_r238903215
 
 

 ##########
 File path: scala-package/infer/src/main/scala/org/apache/mxnet/infer/Classifier.scala
 ##########
 @@ -110,16 +111,20 @@ class Classifier(modelPathPrefix: String,
   : IndexedSeq[IndexedSeq[(String, Float)]] = {
 
     // considering only the first output
-    val predictResultND: NDArray = predictor.predictWithNDArray(input)(0)
+    val predictResultND: NDArray =
 
 Review comment:
   Nit: A comment explaining that we're preemptively copying to the cpu because all the post-processing is happening on the cpu wouldn't hurt.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services