工具实现的融合功能
当前该工具主要实现的为Conv+BN融合:AMCT在量化前会对模型中的"torch.nn.Conv2d+torch.nn.BatchNorm2d"结构做Conv+BN融合,融合后的torch.nn.BatchNorm2d层会被删除。
父主题: 参考信息
当前该工具主要实现的为Conv+BN融合:AMCT在量化前会对模型中的"torch.nn.Conv2d+torch.nn.BatchNorm2d"结构做Conv+BN融合,融合后的torch.nn.BatchNorm2d层会被删除。