Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.nn.parallel.distributed.DistributedDataParallel Class Reference
Inheritance diagram for torch.nn.parallel.distributed.DistributedDataParallel:

Public Member Functions

def __init__ (self, module, device_ids=None, output_device=None, dim=0, broadcast_buffers=True, process_group=None, bucket_cap_mb=25, check_reduction=False)
def __getstate__ (self)
def __setstate__ (self, state)
def forward (self, inputs, kwargs)
def scatter (self, inputs, kwargs, device_ids)
def parallel_apply (self, replicas, inputs, kwargs)
def gather (self, outputs, output_device)
def train (self, mode=True)

Public Attributes


Detailed Description

Definition at line 19 of file

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