torch.nn.parallel.distributed_cpu.DistributedDataParallelCPU Class Reference

Inheritance diagram for torch.nn.parallel.distributed_cpu.DistributedDataParallelCPU:

Public Member Functions
def	__init__ (self, module)
def	sync_parameters (self)
def	forward (self, inputs, kwargs)

Public Attributes
	module
	needs_reduction

Detailed Description

Definition at line 10 of file distributed_cpu.py.

---

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

• torch/nn/parallel/distributed_cpu.py