LazyBatchNorm1d

class torch.nn.modules.batchnorm.LazyBatchNorm1d(eps=1e-05, momentum=0.1, affine=True, track_running_stats=True, device=None, dtype=None, *, bias=True) [source]

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

Отложенная инициализация на основе аргумента num_features класса BatchNorm1d, значение которого определяется по 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 [source]

псевдоним BatchNorm1d