pytorch
makcooo
这个作者很懒,什么都没留下…
展开
-
tensorrt踩坑日记(python pytorch 转 onnx 推理加速)
配置:Ubuntu 16.04python 3.6onnx 1.6pytorch 1.5pycuda 2019.1.2torchvision 0.1.8建议详读,先安装好环境官网指导手册步骤:1.将pytorch模型转换成onnx模型这边用的是Darknet生成的pytoch模型import torchfrom torch.autograd import Variableimport onnxinput_name = ['input']output_n原创 2020-08-01 16:59:47 · 4031 阅读 · 2 评论 -
pytorch YOLO6D训练报错总结
1.CuDNN error: CUDNN_STATUS_SUCCESS在前面加入cudnn.benchmark = True原创 2019-08-29 11:04:24 · 821 阅读 · 0 评论