This repository was archived by the owner on Nov 17, 2023. It is now read-only.
[MXNET-1249] Fix Object Detector Performance with GPU #13522
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
piyushghai
reviewed
Dec 4, 2018
| : IndexedSeq[IndexedSeq[(String, Array[Float])]] = { | ||
|
|
||
| val predictResult = predictor.predictWithNDArray(input)(0) | ||
| val predictResult = predictor.predictWithNDArray(input)(0).asInContext(Context.cpu()) |
Contributor
There was a problem hiding this comment.
If possible, can we apply this to ImageClassifier class as well ?
Maybe this can provide a speed boost there as well ?
Contributor
There was a problem hiding this comment.
Same nitpick I had on imageclassifier. A comment explaining this wouldn't hurt since it's not obvious why we're doing this.
zachgk
approved these changes
Dec 4, 2018
piyushghai
approved these changes
Dec 5, 2018
Contributor
piyushghai
left a comment
piyushghai
left a comment
There was a problem hiding this comment.
Looks ok to me. 💯 🥇
Please also add comments where you are the parArray. That's a nice optimization that might not be very obvious while reading the code :)
andrewfayres
reviewed
Dec 5, 2018
scala-package/infer/src/main/scala/org/apache/mxnet/infer/Classifier.scala
Show resolved
Hide resolved
| : IndexedSeq[IndexedSeq[(String, Array[Float])]] = { | ||
|
|
||
| val predictResult = predictor.predictWithNDArray(input)(0) | ||
| val predictResult = predictor.predictWithNDArray(input)(0).asInContext(Context.cpu()) |
Contributor
There was a problem hiding this comment.
Same nitpick I had on imageclassifier. A comment explaining this wouldn't hurt since it's not obvious why we're doing this.
TaoLv
added a commit
that referenced
this pull request
Dec 6, 2018
…icense file" (#13558) * Revert "Chi_square_check for discrete distribution fix (#13543)" This reverts commit cf6e8cb. * Revert "Updated docs for randint operator (#13541)" This reverts commit e0ff3c3. * Revert "Simplifications and some fun stuff for the MNIST Gluon tutorial (#13094)" This reverts commit 8bbac82. * Revert "Fix #13521 (#13537)" This reverts commit f6b4665. * Revert "Add a retry to qemu_provision (#13551)" This reverts commit f6f8401. * Revert "[MXNET-769] Use MXNET_HOME in a tempdir in windows to prevent access denied due t… (#13531)" This reverts commit bd8e0f8. * Revert "[MXNET-1249] Fix Object Detector Performance with GPU (#13522)" This reverts commit 1c8972c. * Revert "Fixing a 404 in the ubuntu setup doc (#13542)" This reverts commit cb0db29. * Revert "Bumped minor version from 1.4.0 to 1.5.0 on master, updated License file (#13478)" This reverts commit 40db619.
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
4 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
Found Object Detector contains performance problem with GPU which is slower than CPU. The root cause is the long time taken in the post-processing.
@piyushghai @andrewfayres @zachgk
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.