From e657b893412550bfbc6e9588ed1958e656fc6737 Mon Sep 17 00:00:00 2001 From: "Joshua Z. Zhang" Date: Thu, 28 Sep 2017 10:05:22 -0700 Subject: [PATCH] revert v1 (#53) --- nnvm/tutorials/from_mxnet.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/nnvm/tutorials/from_mxnet.py b/nnvm/tutorials/from_mxnet.py index 7068020f15de0..7fb96f166797e 100644 --- a/nnvm/tutorials/from_mxnet.py +++ b/nnvm/tutorials/from_mxnet.py @@ -28,7 +28,7 @@ from mxnet.gluon.utils import download import Image from matplotlib import pyplot as plt -block = get_model('resnet18_v2', pretrained=True) +block = get_model('resnet18_v1', pretrained=True) img_name = 'cat.jpg' synset_url = ''.join(['https://gist.githubusercontent.com/zhreshold/', '4d0b62f3d01426887599d4f7ede23ee5/raw/', @@ -103,12 +103,12 @@ def block2symbol(block): return sym, args, auxs mx_sym, args, auxs = block2symbol(block) # usually we would save/load it as checkpoint -mx.model.save_checkpoint('resnet18_v2', 0, mx_sym, args, auxs) -# there are 'resnet18_v2-0000.params' and 'resnet18_v2-symbol.json' on disk +mx.model.save_checkpoint('resnet18_v1', 0, mx_sym, args, auxs) +# there are 'resnet18_v1-0000.params' and 'resnet18_v1-symbol.json' on disk ###################################################################### # for a normal mxnet model, we start from here -mx_sym, args, auxs = mx.model.load_checkpoint('resnet18_v2', 0) +mx_sym, args, auxs = mx.model.load_checkpoint('resnet18_v1', 0) # now we use the same API to get NNVM compatible symbol nnvm_sym, nnvm_params = nnvm.frontend.from_mxnet(mx_sym, args, auxs) # repeat the same steps to run this model using TVM