Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

SSD300 VGG16 ERROR tvm.error.OpNotImplemented: Operator L2Normalization is not supported in frontend MXNet. #3068

Closed
yjcn opened this issue Apr 22, 2019 · 1 comment

Comments

@yjcn
Copy link

yjcn commented Apr 22, 2019

I have trained a ssd model by using mxnet_ssd. I deployed the model,and the model works well in mxnet.And I want use the tvm(git clone from github and build with LLVM ). But there still is a error: tvm.error.OpNotImplemented: Operator L2Normalization is not supported in frontend MXNet.
I use this code to convert the model.

import mxnet as mx
from mxnet.io import DataBatch, DataDesc
import cv2
from collections import namedtuple
import numpy as np
import tvm
import nnvm

model_prefix='./model/deploy_ssd_vgg16_reduced_300'
epoch=240
#model_prefix='./model/deploy_ssd_mobilenet_v2_300'
#epoch=239
batch_size=1
data_shape=(300,300)
#ctx=mx.gpu(0)
shape_dict = {'data': (1, 3, *data_shape)}
load_symbol, args, auxs= mx.model.load_checkpoint(model_prefix, epoch)

target = tvm.target.create("llvm")

opt_level=3

nnvm_sym, nnvm_params = nnvm.frontend.from_mxnet(load_symbol, args, auxs)
with nnvm.compiler.build_config(opt_level=opt_level):
   graph, lib, params = nnvm.compiler.build(nnvm_sym, target, shape_dict, params=nnvm_params)
lib.export_library("./deploy_lib.so")
print('lib export succeefully')
with open("./deploy_graph.json", "w") as fo:
   fo.write(graph.json())
with open("./deploy_param.params", "wb") as fo:
   fo.write(nnvm.compiler.save_param_dict(params))

image
I knew someone met the same problembefore (issue 1223). And it looks like this problem has been fixed.But I still meet the error.
I also test the deploy_ssd_mobilenet_v2_300 model , It works well,and there is no error.
The tvm version is 0.6.dev, the nnvm version is 0.8.0
image
Does anyone know how to solve it?

@tqchen
Copy link
Member

tqchen commented Apr 22, 2019

For this kind of community questions, please open a new thread on https://discuss.tvm.ai/

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants