ruabuliuqiu

I have the same problem,it need architure, as follows: 【My code】 import torch from torch.autograd import Variable from backbone.resnet import get_pose_net from backbone.vovnetv2 import VoVNet39_slim output_onnx = 'centernet.onnx' pytorch_model =...

> I have the same problem,how to modify it。。 > torch.onnx need both the architure and the parameters. You should define the network before load xxx.pth. Yes，you are right, acording...

【My problem is as follows】： Traceback (most recent call last): File "convert1.py", line 11, in torch.onnx.export(model, dummy_input, "alexnet.onnx", verbose=False, input_names=input_names, output_names=output_names) File "/root/anaconda3/lib/python3.6/site-packages/torch/onnx/__init__.py", line 168, in export custom_opsets, enable_onnx_checker, use_external_data_format)...

[https://github.com/QuasarLight/Pytorch_Face_Recognition-/blob/master/Utils/Datasets_Utils/generate_dataset_list.py](url) It works，but first you need extract images from webface(train.rec)

> train_list 就是一个分类数据的list，每个类别后面给一个类别编号即可，你只要训练过分类模型应该就知道 xiexie

> i am so sorry to ask for this easy question..... > but i dont know what the labels of casia webface dataset is..... and i dont know how to...

> I tried to disable the Cudnn when compile Caffe, and then I can convert the model without this problem. Yes，you are right! Good!