from torch.quantization.fx.fusion_patterns import ConvBNReLUFusion, ModuleReLUFusion

AntigoneRandy / QuantBackdoor_EFRAP

Apache License 2.0

7 stars 0 forks source link

from torch.quantization.fx.fusion_patterns import ConvBNReLUFusion, ModuleReLUFusion #2

Closed bieyl closed 2 months ago

bieyl commented 2 months ago

ImportError: cannot import name "ConvBNReLUFusion“ from "torch.quantization.fx.fusion_patterns 请问是不是mqbench和pytorch版本不匹配

AntigoneRandy commented 2 months ago

Hi dear @bieyl, thanks for your interest in our work. It is indeed possibly beacuse of the version mismatch between torch and mqbench. Similar problem was previously reported in this issue, and you may refer to it for possible solutions.