python3 train.py
python3 export_onnx.py
mkdir build && cd build
cmake ..
make -j
./lenet_trt ../config.yaml ../samples
model | PyTorch | TensorRT |
---|---|---|
inference time | 0.4ms | 0.046ms |
Name | Name | Last commit date | ||
---|---|---|---|---|
parent directory.. | ||||
python3 train.py
python3 export_onnx.py
mkdir build && cd build
cmake ..
make -j
./lenet_trt ../config.yaml ../samples
model | PyTorch | TensorRT |
---|---|---|
inference time | 0.4ms | 0.046ms |