Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
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

- Public Attributes inherited from torch.nn.modules.batchnorm._BatchNorm

Detailed Description

Definition at line 322 of file

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