Skip to content

modricwang/Pytorch-Model-to-TensorRT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch-Model-to-TensorRT

Convert a Pytorch model to TensorRT engin (optional: int8 engin)

Required:

Python packages: in requirements.txt

External:

  • CUDA == 9.0
  • CUDNN == 7.3.1
  • TensorRT == 4.0.2.6 link

Data And Checkpoint Prepare

You should prepare data and model weights in like this way:

project  
│
└───data
│   │   test.txt
│   │   fake-test.txt
│   │   images
│   
└───checkpoint
    │   model_best.pth

The list of test file should follow this format:

file0,label0
file1,label1
... ...

Or you can download the example from BaiduYun

Convert step:

python main.py

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages