线性层量化失败

时间:2019-10-17 17:13:14

标签: pytorch

我用线性输出对MobileNetV2进行了量化。将Conv + BN + Relu融合到ConvBnReLU2d,将Linear + Relu融合到LinearReLU层。执行它时,出现错误信息:

找不到要分配给运算符'quantized :: linear_relu'的内核。

所有转换层似乎都能正常工作。根据文档,支持LinearReLU: https://pytorch.org/docs/stable/quantization.html#torch.nn.intrinsic.LinearReLU

0 个答案:

没有答案