
TRT
TensorRT从入门到放弃,学习笔记,夹带私货。
麦克斯韦恶魔
我今天就是要殉爆胡德酱
展开
-
TensorRT实战(二) 如何使用TRT python API搭建简单的VGG16网络
如何使用TRT python API搭建简单的VGG16网络,超级详细,内附代码、简单报错解决办法,持续更新,未完待续。原创 2020-01-01 17:45:08 · 9746 阅读 · 22 评论 -
TensorRT实战(一) 如何搭建Batch Normalization层
TensorRT实战(一) 如何搭建Batch Normalization层文章目录TensorRT实战(一) 如何搭建Batch Normalization层PyTorch的Batch NormalizationTRT API实现fused Batch Normalization参考资料本文使用TensorRT Python API进行搭建,C++ API搭建方法异曲同工,使用的BN lay...原创 2019-12-12 19:42:23 · 3959 阅读 · 26 评论 -
PyTorch2ONNX2TensorRT 踩坑日志
从“用PyTorch写的网络,通过ONNX,使用TensorRT序列化,最终完成模型加速”的全流程踩坑日志。持续更新。原创 2019-12-07 15:30:05 · 21438 阅读 · 26 评论