NNI模型加速：ValueError: Modules that have backward hooks assigned can‘t be compiled:

概要：采用nni进行Speed Up的时候，报错ValueError: Modules that have backward hooks assigned can't be compiled: Conv2d(3, 18, kerne）

问题定位：
错误提示含义大致为模型有backward hook，无法编译。定位到代码里：

self.p_conv.register_backward_hook(self._set_lr)

代码原形为torch.nn.Module.register_backward_hook，针对nn.Module这个对象，作用为提取中间层参数，比如梯度之类的。详细介绍参考pytorch笔记：pytorch中的钩子（Hook）有何作用？_Activewaste-CSDN博客_hook pytorch

解决方法：
只能是注释掉这句代码了，好在对我的模型没啥影响。其实把问题的产生原因讲清楚是最好的，奈何当前能力有限，只能先搁置了。

NNI模型加速：ValueError: Modules that have backward hooks assigned can‘t be compiled:相关推荐

NNI模型剪枝压缩使用的较详细记录
使用微软nni工具,基于pytorch进行模型压缩记录总结,最后的结论一很久以前实验的,可能会有点误差,或者NNI工具更新了.但是nni模型剪枝压缩的整体流程是这样的,在此简单记录一下. 一.模型的剪 ...
sklearn随机森林模型：ValueError: Unknown label type: ‘unknown‘
sklearn随机森林模型:ValueError: Unknown label type: 'unknown' 目录 sklearn随机森林模型:ValueError: Unknown label t ...
10倍加速！爱奇艺超分辨模型加速实践
关注公众号,发现CV技术之美随着终端播放设备的升级,观众对于视频的品质需求也逐步提升.需求从最开始的高清过渡到4K,最近8K也有开始流行的趋势.除了对于分辨率提升的需求之外,视频在采集的过程中,也难 ...
【模型加速】关于模型加速的总结
概述 ● 模型加速的目标: a. Increase inference speed:加快推理速度(应用层面). b. Reduce model size:压缩模型. ● 关于模型的加速大致可以分为三个 ...
yolo模型部署——tensorRT模型加速+triton服务器模型部署
将最近的工作做个记录,方便日后学习回顾: 1.针对项目需求开发满足任务的模型,拿到任务就要去选相应的算法,由于是工程应用型,必须找填坑多的算法,这样遇到问题可参考的资料多. 2.做好以后,还要将开发的 ...
模型加速之轻量化网络
模型加速之轻量化网络当前物体检测结构大都依赖使用卷积网络进行特征提取,即 Backbone,通常使用 VGGNet.ResNet 等优秀的基础网络,但是这些网络往往计算量巨大,依赖这些基础网络的检测 ...
模型转换、模型压缩、模型加速工具汇总
点击上方"计算机视觉工坊",选择"星标" 干货第一时间送达编辑丨机器学习AI算法工程一.场景需求解读在现实场景中,我们经常会遇到这样一个问题,即某篇论 ...
Ultralytics公司YOLOv8来了(训练自己的数据集并基于NVIDIA TensorRT和华为昇腾端到端模型加速)--跟不上“卷“的节奏
Official YOLOv8 训练自己的数据集并基于NVIDIA TensorRT和华为昇腾端到端模型加速说明: 本项目支持YOLOv8的对应的package的版本是:ultralytics-8. ...
win10下 yolov8 tensorrt模型加速部署【实战】
Windows10下yolov8 tensorrt模型加速部署[实战] TensorRT-Alpha基于tensorrt+cuda c++实现模型end2end的gpu加速,支持win10.linux ...

NNI模型加速：ValueError: Modules that have backward hooks assigned can‘t be compiled:

NNI模型加速：ValueError: Modules that have backward hooks assigned can‘t be compiled:相关推荐

最新文章

热门文章