search

vadimkantorov / caffemodel2pytorch

Convert Caffe models to PyTorch
389 stars 84 forks source link

How to use converted-model (pt)?? #10

Closed edwardcho closed 5 years ago

edwardcho commented 5 years ago

Hello. My caffemodel was changed pt-model (pytorch model) using your convertor. But, I do not know how to use it.

I think that inference or training of pytorch should need two files.

  1. definition python code.
  2. pt model

If you can, please upload sample code (inference code and training)

vadimkantorov commented 5 years ago

You are right. You would need Python code as well.

Unfortunately, I don't work on this codebase anymore, so can't provide any additional examples. The example with OICR conversion in README should be sufficient. You get a nn.Module model object, and then train/infer in PyTroch as usual.

Good luck!