模型压缩
yangdeshun888
本人QQ:2891720537,欢迎添加咨询问题。
展开
-
centos7 安装TensorRT
参考博客:centos7 安装TensorRT 注意: a、主要两个文件:TensorRT、pycuda包文件。 b、tensorRT需要到官网下载后解压到固定的系统文件夹下即可。pycuda则是直接使用pip安装即可。 c、加速流程:pytorch模型转换成onnx,使用自身的onnx模块即可;然后通过onnx转换成tensorrt的模型格式,大概后缀为.engine;最后就可以通过调用这个engine进行相应的加速使用。 1.安装cuda、cudnn 问题: patchlevel.h没有那个文件 将py原创 2021-03-15 15:32:53 · 1292 阅读 · 0 评论 -
深度学习知识六:(模型量化压缩)----pytorch自定义Module,并通过其理解DoReFaNet网络定义方法。
参考中文官方,详情参考:PyTorch 如何自定义 Module 1.自定义Module Module 是 pytorch 组织神经网络的基本方式。Module 包含了模型的参数以及计算逻辑。Function 承载了实际的功能,定义了前向和后向的计算逻辑。 下面以最简单的 MLP 网络结构为例,介绍下如何实现自定义网络结构。完整代码可以参见repo。 1.1 Function Function ...原创 2020-03-20 14:48:05 · 1190 阅读 · 0 评论