Skip to content

Files

Latest commit

 

History

History
 
 

lenet

LeNet Network PyTorch=>ONNX=>TensorRT

1.Reference

2.Train PyTorch Network

python3 train.py

3.Export ONNX Model

python3 export_onnx.py

4.Build lenet_trt Project

mkdir build && cd build
cmake ..
make -j

5.Run lenet_trt

./lenet_trt ../config.yaml ../samples

6.Benchmark(2080Ti)

model PyTorch TensorRT
inference time 0.4ms 0.046ms