Replicates a model on different GPUs. Specifically, this function implements single-machine multi-GPU data parallelism. It works in the following way: - Divide the model's input(s) into multiple sub-batches. - Apply a model copy on each sub-batch. Every model copy is executed on a de
(model, gpus, cpu_merge=True, cpu_relocation=False)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected