我用线性输出对MobileNetV2进行了量化。将Conv + BN + Relu融合到ConvBnReLU2d,将Linear + Relu融合到LinearReLU层。执行它时,出现错误信息:
找不到要分配给运算符'quantized :: linear_relu'的内核。
所有转换层似乎都能正常工作。根据文档,支持LinearReLU: https://pytorch.org/docs/stable/quantization.html#torch.nn.intrinsic.LinearReLU