Hi, I am trying QAT for YOLOv5 and this is the 1st portion of the model

image834×269 37.6 KB

Using:
torch.quantization.fuse_modules(module, [["conv","bn","relu"]],inplace=True)
the model does not seemed fused

image1590×699 119 KB

Whereas if i use:
torch.quantization.fuse_modules(module, [["conv","bn"]],inplace=True)
the model is fused

image1601×683 111 KB

Is this a bug?