Method forward

prototype/utils/dist.py:60–61

(self, *inputs, **kwargs)

            self._register_hooks()

    def forward(self, *inputs, **kwargs):
        return self.module(*inputs, **kwargs)

    def _register_hooks(self):
        for i, (name, p) in enumerate(self.named_parameters()):