Pytorch2TensorRT

This is a demo for pytorch => onnx => tensorrt

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
augmentation		augmentation
data		data
models		models
utils		utils
calc_errors.py		calc_errors.py
onnx2engine.py		onnx2engine.py
pytorch2onnx.py		pytorch2onnx.py
pytorch_infer.py		pytorch_infer.py
readme.md		readme.md
requirements.txt		requirements.txt
train.py		train.py
trt_infer.py		trt_infer.py

Provide feedback