def __init__(self, proc_id, pipe, master):
    super(_Worker, self).__init__()
    self.proc_id = proc_id
    self.pipe = pipe
    self.converter = master.converter
    self.model = master._master
    self.device = master._devices[proc_id]
    self.iterator = master._mpu_iterators[proc_id]
    self.n_devices = len(master._devices)
def __init__(self, proc_id, pipe, master):
    super(_Worker, self).__init__()
    self.proc_id = proc_id
    self.pipe = pipe
    self.converter = master.converter
    self.model = master._master
    self.device = master._devices[proc_id]
    self.iterator = master._mpu_iterators[proc_id]
    self.n_devices = len(master._devices)