Skip to content

Latest commit

 

History

History
28 lines (23 loc) · 669 Bytes

README.md

File metadata and controls

28 lines (23 loc) · 669 Bytes

torchtrtz

This project is mainly used t generate tensorrt weights for makaveli10/cpptensorrtz

Getting Started

  1. Tested with python==3.7.9
  2. Install torch, torchvision
 $ pip install torch==1.6.0
 $ pip install torchvision==0.7.0
  1. Install torchsummary
 $ pip install torchsummary

How to Run

All the models are from torchvision. model.py will download and save the torch weights. Then gen_wts.py will write the pytorch model in a "vgg16".wts file as required by TensorRT.

Example VGG:

 $ cd vgg
 $ python models.py
 $ python gen_trtwts.py