Skip to content
Branch: master
Find file History
Latest commit dc512d6 May 29, 2019
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
PyTorch_Inference.ipynb PyTorch inference using ONNX and Caffe2 May 28, 2019
PyTorch_to_ONNX_for_Inference.ipynb
README.md cleaning up May 29, 2019
environment.yml
giraffe.jpg

README.md

PyTorch Model Inference using ONNX and Caffe2

We show how to do inference with a PyTorch model using ONNX and Caffe2.

Dependencies

torch
torchvision
onnx

Create conda environment

conda env create -f environment.yml

Download models

The required models can be downloaded from the following links: Animals Caltech

Activate conda environment

conda activate pytorch_inference

Outline

The enclosed notebook lets you

  1. Export the PyTorch .pt model to ONNX model
  2. Use the ONNX model for inference in Caffe2

More details can be found in the blog

You can’t perform that action at this time.