当前该工具主要实现的是BN融合功能,具体如下:
Conv+BN融合:AMCT在量化前会对模型中的"nn.Conv2d+nn.BatchNorm2d"结构做Conv+BN融合,融合后的"nn.BatchNorm2d"层会被删除。