下载
中文
注册

工具实现的融合功能

当前该工具主要实现的是BN融合功能,具体如下:

Conv+BN融合:昇腾模型压缩工具在量化前会对模型中的 "nn.Conv2d+nn.BatchNorm2d" 结构做Conv+BN融合,融合后的 "nn.BatchNorm2d" 层会被删除。