就是我看量化代码里有对Conv2dStaticSamePadding这一层的量化接口,但是在我的网络的代码里又自定义了一个实现Conv2dStaticSamePadding的类(里面调用了nn.conv2d),可能二者产生了冲突,然后我把网络代码里的Conv2dStaticSamePadding类改了个名,全都改成了Conv2dStaticSamePaddings,就可以量化了,但是加载量化模型时就报上面那个错误了,nn.conv2d跑不上去了展开
就是我看量化代码里有对Conv2dStaticSamePadding这一层的量化接口,但是在我的网络的代码里又自定义了一个实现Conv2dStaticSamePadding的类(里面调用了nn.conv2d),可能二者产生了冲突,然后我把网络代码里的Conv2dStaticSamePadding类改了个名,全都改成了Conv2dStaticSamePaddings,就可以量化了,但是加载量化模型时就报上面那个错误了,nn.conv2d跑不上去了展开
请将量化后的模型 以及load 之后的模型打印出来,看一下是量化后的问题,还是load的问题,量化正确的模型conv2d会被替换成MLUconver2d ,理论上不会出现conver2d这个算子,详情请参考官方文档
请登录后评论