MCPcopy Create free account
hub / github.com/ajbrock/BigGAN-PyTorch / execute_replication_callbacks

Function execute_replication_callbacks

sync_batchnorm/replicate.py:27–47  ·  view source on GitHub ↗

Execute an replication callback `__data_parallel_replicate__` on each module created by original replication. The callback will be invoked with arguments `__data_parallel_replicate__(ctx, copy_id)` Note that, as all modules are isomorphism, we assign each sub-module with a context

(modules)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

replicateMethod · 0.85
new_replicateFunction · 0.85

Calls 2

CallbackContextClass · 0.85

Tested by

no test coverage detected