Sample help better understand trt
mkdir build && cd build
cmake -DCMAKE_CUDA_FLAGS="--expt-extended-lambda -std=c++14" ..
./export ../pretrained/mnist.onnx ../pretrained/mnist.plan
./infer pretrained/mnist.plan data/9.pgm
thaint2901/sample_onnx
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|