![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
模型压缩
文章平均质量分 67
Coding-Prince
一个不知名的元婴初期炼丹师!
不再炼丹!!!
展开
-
NNI speedup_model()代码笔记
前情提要:NNI是微软开发的调参工具,功能有很多,这里介绍其中的一个分支-模型压缩。模型压缩流程:1.模型prune2.模型speedup模型prune不多介绍,模型speedup代码基本就这三行,不过一般运行会出很多问题,除非speedup的模型非常简单且常规 apply_compression_results(net, masks_file, device) m_speedup = ModelSpeedup(net, dummy_input, masks_file, device)原创 2022-03-30 20:38:01 · 4803 阅读 · 0 评论 -
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)确实有这个东西,但是解决方法:只能是注释掉这句代码了,好在对我的模型没原创 2022-03-14 18:31:54 · 2473 阅读 · 1 评论