MCPcopy Create free account
hub / github.com/csxmli2016/DFDNet / DataParallelWithCallback

Class DataParallelWithCallback

sync_batchnorm/replicate.py:50–67  ·  view source on GitHub ↗

Data Parallel with a replication callback. An replication callback `__data_parallel_replicate__` of each module will be invoked after being created by original `replicate` function. The callback will be invoked with arguments `__data_parallel_replicate__(ctx, copy_id)` Example

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

convert_modelFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected