Rate this Page
★★★★★

LazyBatchNorm3d#

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

Atorch.nn.BatchNorm3d module with lazy initialization.

Lazy initialization is done for thenum_features argument of theBatchNorm3d that is inferredfrom theinput.size(1).The attributes that will be lazily initialized areweight,bias,running_mean andrunning_var.

Check thetorch.nn.modules.lazy.LazyModuleMixin for further documentationon lazy modules and their limitations.

Parameters

  • eps (float) – a value added to the denominator for numerical stability.Default: 1e-5

  • momentum (Optional[float]) – the value used for the running_mean and running_varcomputation. Can be set toNone for cumulative moving average(i.e. simple average). Default: 0.1

  • affine (bool) – a boolean value that when set toTrue, this module haslearnable affine parameters. Default:True

  • track_running_stats (bool) – a boolean value that when set toTrue, thismodule tracks the running mean and variance, and when set toFalse,this module does not track such statistics, and initializes statisticsbuffersrunning_mean andrunning_var asNone.When these buffers areNone, this module always uses batch statistics.in both training and eval modes. Default:True

cls_to_become[source]#

alias ofBatchNorm3d