def setup(self):
    (_, comm_id) = self.pipe.recv()
    self.comm = nccl.NcclCommunicator(self.n_devices, comm_id, self.proc_id)
    self.model.to_gpu(self.device)
    self.reporter = reporter.Reporter()
    self.reporter.add_observer('main', self.model)
    self.reporter.add_observers('main', self.model.namedlinks(skipself=True))
def setup(self):
    (_, comm_id) = self.pipe.recv()
    self.comm = nccl.NcclCommunicator(self.n_devices, comm_id, self.proc_id)
    self.model.to_gpu(self.device)
    self.reporter = reporter.Reporter()
    self.reporter.add_observer('main', self.model)
    self.reporter.add_observers('main', self.model.namedlinks(skipself=True))