search

longcw / pytorch2caffe

Convert PyTorch model to Caffemodel
541 stars 151 forks source link

Installation requirements #1

Closed apereiracv closed 7 years ago

apereiracv commented 7 years ago

Hi, I am thinking to use this to convert resnet-18/34 models to caffe. Does your port need complicated installation from source or can I just pip install the libraries needed in your demo file?

longcw commented 7 years ago

You may need pytorch, caffe and pycaffe installation.