Uh oh!
There was an error while loading. Please reload this page.
[DARKNET]RNN Support for darknet - #1443
Conversation
siju-samuel
commented
Jul 17, 2018
@PariksheetPinjari909 , @srkreddy1238@Huyuwei@masahi |
| target = 'llvm' | ||
| shape_dict = {'data': data.shape} | ||
| #with nnvm.compiler.build_config(opt_level=2): |
There was a problem hiding this comment.
remove the commented line if not required.
| shape_dict = {'data': data.shape} | ||
| #with nnvm.compiler.build_config(opt_level=2): | ||
| graph, library, params = nnvm.compiler.build(sym, target, shape_dict, dtype, params=params) | ||
| ###################################################################### |
There was a problem hiding this comment.
Correct the comment styles.
| ###################################################################### | ||
| # Prepare cfg and weights file | ||
| # Pretrained model available | ||
| # -------------------------------------------------------------------- |
| CFG_NAME = MODEL_NAME + '.cfg' | ||
| WEIGHTS_NAME = MODEL_NAME + '.weights' | ||
| CFG_URL = 'https://github.com/siju-samuel/darknet/blob/master/cfg/' + \ |
There was a problem hiding this comment.
may use dmlc/web-data instead
| ###################################################################### | ||
| # Compile the model on NNVM | ||
| # -------------------------------------------------------------------- |
siju-samuel
commented
Jul 18, 2018
@srkreddy1238 Thanks for the comments. All comments are addressed. Please check again. TIA |
| CFG_URL = REPO_URL + 'cfg/' + CFG_NAME + '?raw=true' | ||
| WEIGHTS_URL = REPO_URL + 'weights/' + WEIGHTS_NAME + '?raw=true' | ||
| def _dl_progress(count, block_size, total_size): |
There was a problem hiding this comment.
Tutorial scripts gets exported to docs.tvm.ai (Ex: https://docs.tvm.ai/tutorials/nnvm/get_started.html) .
Hence suggest to simplify the example by using existing methods.
Here we could use glueon download methods instead of rewriting.
| with nnvm.compiler.build_config(opt_level=2): | ||
| graph, lib, params = nnvm.compiler.build(sym, target, shape_dict, dtype_dict, params) | ||
| # Save the json |
There was a problem hiding this comment.
Saving lib already explained in get_started.py. Can leave a doc pointer instead.
srkreddy1238
commented
Jul 18, 2018
@siju-samuel with above comments LGTM. |
srkreddy1238
left a comment
There was a problem hiding this comment.
@siju-samuel thanks. good to go now.
siju-samuel
commented
Jul 19, 2018
@srkreddy1238 Thanks for the review. |
tqchen
commented
Jul 19, 2018
Thanks @siju-samuel@srkreddy1238@PariksheetPinjari909 , this is now merged |
RNN Support for DARKNET frontend is added.
This PR consists of
Thanks for contributing to TVM! Please refer to guideline https://docs.tvm.ai/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from others in the community.