donnyyou / torchcv

TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision
https://pytorchcv.com
Apache License 2.0
2.25k stars 378 forks source link

将sfnet训练出来的pth模型转换成onnx模型应该怎么做 #163

Open Truring opened 4 years ago

Truring commented 4 years ago

如果需要将pth模型转换成onnx模型,需要先创建一个model对象,model对象应该怎么创建呢?下图中的写法会报错 image image

Truring commented 4 years ago

@donnyyou 麻烦您指点一下我上面提的问题,非常感谢!