[RELAY][DOCS] Port from_mxnet tutorial to relay #2608
Merged
Conversation
This file contains 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
eqy
force-pushed
the
relay_from_mxnet
branch
from
447525f to
349eb48
Compare
zhreshold
suggested changes
Feb 16, 2019
tutorials/relay/from_mxnet.py
Outdated
| target = 'cuda' | ||
| shape_dict = {'data': x.shape} | ||
| with relay.build_config(opt_level=3): | ||
| graph, lib, params = relay.build_module.build(net, target, params=params) |
There was a problem hiding this comment.
I think following is the recommended way to inference in Relay
with relay.build_config(opt_level=1):
intrp = relay.build_module.create_executor('graph', sym, tvm.gpu(0), target)
tvm_output = intrp.evaluate(sym)(tvm.nd.array(x.astype(dtype)), **params).asnumpy()There was a problem hiding this comment.
thanks for the suggestion; looks like the output of the interpreter is already numpy?
yzhliu
approved these changes
Feb 16, 2019
|
Thanks @eqy @zhreshold This is merged. |
libing4752
pushed a commit
to libing4752/tvm
that referenced
this pull request
Feb 18, 2019
* check in * update build and run
wweic
pushed a commit
to neo-ai/tvm
that referenced
this pull request
Feb 20, 2019
* check in * update build and run
wweic
pushed a commit
to neo-ai/tvm
that referenced
this pull request
Feb 20, 2019
* check in * update build and run
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
Port the prediction part of the NNVM
from_mxnettutorial to relay (note that the checkpointing part is omitted for now).