search

warmspringwinds / pytorch-cpp

Pytorch C++ Library
369 stars 68 forks source link

Support of customized models #20

Open DianCh opened 6 years ago

DianCh commented 6 years ago

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!

walktree commented 5 years ago

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