Method hook

prototype/utils/dist.py:72–73

(*ignore)

    def _make_hook(self, name, p, i):
        def hook(*ignore):
            link.allreduce_async(p.grad.data)
        return hook

    def sync_gradients(self):