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