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 должны находиться в режиме eval, а для bn должны быть вычислены текущие буферы.