pytorch-cpp icon indicating copy to clipboard operation
pytorch-cpp copied to clipboard

Published 20 hours ago • verified publisher icon warmspringwinds

Support of customized models

Open DianCh opened this issue 6 years ago • 1 comments

Hi! Does pytorch-cpp support any conversion from customized models built in pytorch, or does it only support standard models (e.g. vgg, res-net) and your own segmentation models? Big thanks!

DianCh avatar Jul 05 '18 01:07 DianCh

Yes, customized models are supported by libtorch. Actually, you could build any model since it has same interface as python version, e.g. I just migrated YOLO v3 in cpp api: https://github.com/walktree/libtorch-yolov3

walktree avatar Dec 05 '18 07:12 walktree