pytorch - 线性层量化失败

Question

我用线性输出量化了一个 MobileNetV2。Conv+BN+Relu 融合到 ConvBnReLU2d，Linear+Relu 融合到 LinearReLU 层。执行它时，我收到错误消息：

没有为操作员 'quantized::linear_relu' 找到要分派的内核。

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

标签： pytorch