[TensorRT] TensorRT官方例程SampleOnnxMNIST代码完全解读!
本文首发于https://yangzhenyu.tech/作者:Frank导读:本文主要带来对TensorRT中自带的sample:sampleOnnxMNIST的源码解读,官方例程是非常好的学习资料,通过吃透一个官方例程,就可以更加深刻地了解TensorRT的每一步流程,明白其中套路,再去修改代码推理我们自己的网络就是很容易的事情了。TensorRT的一份源码要怎么看?TensorRT是一个很具有操作流程性质的代码,我们从main函数开始着手,顺着执行的流程步骤向前推,遇到各种功能的嵌套,就像
原创
2020-12-23 13:04:33 ·
7590 阅读 ·
9 评论