LazyBatchNorm3d

class torch.nn.modules.batchnorm.LazyBatchNorm3d(eps=1e-05, momentum=0.1, affine=True, track_running_stats=True, device=None, dtype=None, *, bias=True) [исходный код]

Модуль torch.nn.BatchNorm3d с отложенной инициализацией.

Отложенная инициализация выполняется для аргумента num_features объекта BatchNorm3d, который определяется по input.size(1). Атрибуты, которые будут инициализированы отложенно: weight, bias, running_mean и running_var.

Дополнительную документацию об отложенно инициализируемых модулях и их ограничениях см. в описании torch.nn.modules.lazy.LazyModuleMixin.

Параметры:

• eps (float) – значение, добавляемое к знаменателю для численной устойчивости. По умолчанию: 1e-5
• momentum (float | None) – значение, используемое при вычислении running_mean и running_var. Для вычисления кумулятивного скользящего среднего (то есть простого среднего) можно задать None. По умолчанию: 0.1
• affine (bool) – логическое значение; если задано True, модуль имеет обучаемые аффинные параметры. По умолчанию: True
• track_running_stats (bool) – логическое значение; если задано True, модуль отслеживает скользящие среднее и дисперсию; если задано False, модуль не отслеживает эти статистики и инициализирует буферы статистики running_mean и running_var значением None. Если эти буферы имеют значение None, модуль всегда использует статистики текущего пакета данных как во время обучения, так и в режиме оценки. По умолчанию: True
• bias (UninitializedParameter) – Если задано False, слой не будет обучать аддитивное смещение (имеет значение, только если affine равно True). По умолчанию: True

cls_to_become [исходный код]

псевдоним BatchNorm3d