semvec/pypi-packages huggingface dataset:
pytorch2caffe 0.1.0
A package to convert pytorch model to caffe model
License: MIT
Requires Python: >=3.5.2


# pytorch2caffe

A package to convert pytorch model to caffe model. 

## Installation

`pip3 install pytorch2caffe`

or

    git clone https://github.com/woodsgao/pytorch2caffe
    cd pytorch2caffe
    python3 setup.py install

## Tutorials

Note: import torch first then import pytorch2caffe

    import torch
    from torchvision.models import resnet
    from pytorch2caffe import pytorch2caffe

    name='resnet18'
    resnet18=resnet.resnet18()
    resnet18.eval()
    dummy_input=torch.ones([1,3,224,224])
    pytorch2caffe.trans_net(resnet18, dummy_input, name)
    pytorch2caffe.save_prototxt('{}.prototxt'.format(name))
    pytorch2caffe.save_caffemodel('{}.caffemodel'.format(name))