Method parallel_apply

utils/pl_utils.py:253–254

(self, replicas, inputs, kwargs)

        return self.gather(outputs, self.output_device)

    def parallel_apply(self, replicas, inputs, kwargs):
        return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])


class GradientAccumulationScheduler: