MCPcopy Create free account
hub / github.com/ansleliu/LightNetPlusPlus / DataParallelModel

Class DataParallelModel

utils/parallel.py:69–104  ·  view source on GitHub ↗

Implements data parallelism at the module level. This container parallelizes the application of the given module by splitting the input across the specified devices by chunking in the batch dimension. In the forward pass, the module is replicated on each device, and each replica

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected