yasenh / libtorch-yolov5

A LibTorch inference implementation of the yolov5
MIT License
372 stars 114 forks source link

export with torch==1.3.0??? #13

Closed dreamhighchina closed 3 years ago

dreamhighchina commented 3 years ago

how can i export model with torch==1.3.0????

yasenh commented 3 years ago

@dreamhighchina Looks like there are some operations not supported in torch v1.3.0, I would suggest you upgrade to at least 1.5.1