ConvReLU3d

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

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

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

Переменные:

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

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

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

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

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