ConvBnReLU3d

class torch.ao.nn.intrinsic.qat.ConvBnReLU3d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=None, padding_mode='zeros', eps=1e-05, momentum=0.1, freeze_bn=False, qconfig=None) [исходный код]

Модуль ConvBnReLU3d — это модуль, полученный путём объединения Conv3d, BatchNorm3d и ReLU, с модулями FakeQuantize для весов, используемый при обучении с учётом квантования.

Мы объединили интерфейсы torch.nn.Conv3d, torch.nn.BatchNorm3d и torch.nn.ReLU.

Аналогично torch.nn.Conv3d, модули FakeQuantize инициализируются значениями по умолчанию.

Переменные:

weight_fake_quant – модуль фиктивного квантования для весов

forward(input) [исходный код]

Выполняет прямой проход через объединённые Conv3d, BatchNorm3d и ReLU.

classmethod from_float(mod, use_precomputed_fake_quant=False) [исходный код]

Создаёт модуль QAT из модуля с плавающей точкой.