Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
data		data
src		src
.gitignore		.gitignore
Dockerfile		Dockerfile
README.md		README.md
run.sh		run.sh

Repository files navigation

Converter PyTorch -> ONNX -> TensorRT

Run docker container

 1. chmod +x run.sh
 2. ./run.sh

Preparation

In src/models define get_model function to get model.
In src.trt.config define all settings.

Convert model

Run PYTHONPATH=. python src/trt/torch2onnx.py for convert torch -> onnx
Run PYTHONPATH=. python src/trt/onnx2trt.py for convert onnx -> tensorrt
Run PYTHONPATH=. python src/trt/check.py for check conversion

About

Conversion PyTorch -> ONNX -> TensorRT

pytorch tensorrt onnx

Report repository

Releases

No releases published

Packages

No packages published

Languages