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

Class DataParallelCriterion

utils/parallel.py:107–147  ·  view source on GitHub ↗

Calculate loss in multiple-GPUs, which balance the memory usage for Semantic Segmentation. The targets are splitted across the specified devices by chunking in the batch dimension. Please use together with :class:`encoding.parallel.DataParallelModel`. Reference: Hang Z

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