LazyBatchNorm3d

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

A torch.nn.BatchNorm3d module with lazy initialization of the num_features argument of the BatchNorm3d that is inferred from the input.size(1). The attributes that will be lazily initialized are weight, bias, running_mean and running_var.

Check the torch.nn.modules.lazy.LazyModuleMixin for further documentation on lazy modules and their limitations.

Parameters:

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

cls_to_become

Псевдоним для BatchNorm3d