torch.nn.utils.fuse_conv_bn_eval¶
- torch.nn.utils.fuse_conv_bn_eval(conv, bn, transpose=False)[源代码]¶
将卷积模块和 BatchNorm 模块融合到单个新的卷积模块中。
- 参数
conv (torch.nn.modules.conv._ConvNd) – 卷积模块。
bn (torch.nn.modules.batchnorm._BatchNorm) – BatchNorm 模块。
transpose (bool, 可选) – 如果为 True,则转置卷积权重。默认为 False。
- 返回值
融合后的卷积模块。
- 返回类型
torch.nn.modules.conv._ConvNd
注意
conv
和bn
必须都处于评估模式,并且bn
必须计算其运行缓冲区。