torch.nn.modules.batchnorm.SyncBatchNorm Class Reference

Inheritance diagram for torch.nn.modules.batchnorm.SyncBatchNorm:

Public Member Functions
def	__init__ (self, num_features, eps=1e-5, momentum=0.1, affine=True, track_running_stats=True, process_group=None)
def	forward (self, input)
Public Member Functions inherited from torch.nn.modules.batchnorm._BatchNorm
def	__init__ (self, num_features, eps=1e-5, momentum=0.1, affine=True, track_running_stats=True)
def	reset_running_stats (self)
def	reset_parameters (self)
def	forward (self, input)
def	extra_repr (self)

Public Attributes
	process_group
	ddp_gpu_size
Public Attributes inherited from torch.nn.modules.batchnorm._BatchNorm
	num_features
	eps
	momentum
	affine
	track_running_stats
	weight
	bias

Detailed Description

Definition at line 322 of file batchnorm.py.

---

The documentation for this class was generated from the following file:

• torch/nn/modules/batchnorm.py