ConvReLU2d

class torch.ao.nn.intrinsic.qat.ConvReLU2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros', qconfig=None) [исходный код]

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

Мы объединили интерфейсы Conv2d и BatchNorm2d.

Переменные:

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

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

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

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

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