You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I want to use mobilefacenet as a classification model instead of an embedding model, but when I run the compile script as follows, pay attention that the checkpoint is not trimmed after the embedding layer but kept the last fully-connetcted layer, I just want get the prob of which person one face belongs to .
import numpy as np
import argparse
import nnvm.compiler
import nnvm.testing
import tvm
from tvm.contrib import graph_runtime
import mxnet as mx
from mxnet import ndarray as nd
prefix,epoch = "/app/model/model-mobilefacenets/model-y1-softmax", 0
sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, epoch)
image_size = (112, 112)
opt_level = 3
shape_dict = {'data': (1, 3, *image_size)}
target = tvm.target.create("cuda")
# "target" means your target platform you want to compile.
#target = tvm.target.create("llvm -mcpu=broadwell")
nnvm_sym, nnvm_params = nnvm.frontend.from_mxnet(sym, arg_params, aux_params)
the error messages is as follows:
OpNotImplemented: Operator BlockGrad is not supported in frontend MXNet.
What can I do to deal with this error? Please help me out, thank you very much. @szad670401
The text was updated successfully, but these errors were encountered:
I want to use mobilefacenet as a classification model instead of an embedding model, but when I run the compile script as follows, pay attention that the checkpoint is not trimmed after the embedding layer but kept the last fully-connetcted layer, I just want get the prob of which person one face belongs to .
the error messages is as follows:
OpNotImplemented: Operator BlockGrad is not supported in frontend MXNet.
What can I do to deal with this error? Please help me out, thank you very much.
@szad670401
The text was updated successfully, but these errors were encountered: