pytorch
流浪_风筝
这个作者很懒,什么都没留下…
展开
-
warp-CTC(SeanNaren)安装踩坑
https://github.com/SeanNaren/warp-ctc 下载下来,解压 mkdir build cd build cmake …/ make 出错 ,还以为是代码问题,换了5.5版本的gcc成功编译 cd pytorch_binding python setup.py install 出错 报cublas_v2.h: No such file or directory 保证cuda正常安装 .bashrc里的路径添加正确 export WARP_CTC_PATH=/home/dell/s原创 2021-01-10 12:31:00 · 519 阅读 · 0 评论 -
pytorch模型转onnx转ncnn
一、pth转onnx 训练生成好.pth文件后,参照网上的代码转onnx,网上代码如下 import torch import torchvision #define resnet18 model model = torchvision.models.resnet18(pretrained=True) #define input shape x = torch.rand(1, 3, 224, 224) #define input and output nodes, can be customized in原创 2020-05-18 14:38:05 · 5284 阅读 · 1 评论