cosmetic

47da14a0 · Michael Carilli · 8a32e428 · 47da14a0
Commit 47da14a0 authored Jul 02, 2019 by Michael Carilli
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 10 deletions

apex/parallel/distributed.py apex/parallel/distributed.py +9 -10

No files found.
--- a/apex/parallel/distributed.py
+++ b/apex/parallel/distributed.py
@@ -356,7 +356,6 @@ class DistributedDataParallel(Module):
                    grad_acc = param_tmp.grad_fn.next_functions[0][0]
                    def allreduce_hook(*unused):
                        if self.prof:
                            torch.cuda.nvtx.range_push("allreduce_hook")