摘要: 本文是基于TensorRT 5.0.2基础上,关于其内部的fc_plugin_caffe_mnist例子的分析和介绍。 本例子相较于前面例子的不同在于,其还包含cpp代码,且此时依赖项还挺多。该例子展示如何使用基于cpp写的plugin,用tensorrt python 绑定接口和caffe解析器一 阅读全文
posted @ 2019-03-14 19:25 仙守 阅读(2828) 评论(0) 推荐(0) 编辑
摘要: 本文是基于TensorRT 5.0.2基础上,关于其内部的network_api_pytorch_mnist例子的分析和介绍。 本例子直接基于pytorch进行训练,然后直接导出权重值为字典,此时并未dump该权重;接着基于tensorrt的network进行手动设计网络结构并填充权重。本文核心在于 阅读全文
posted @ 2019-03-14 12:19 仙守 阅读(1858) 评论(0) 推荐(0) 编辑