def _parallelize(self):
    if self.multiprocess:
        self.model = convert_sync_bn(self.model).to(self.device)
        self.parallel_model = DistributedDataParallel(self.model, self.gpus, broadcast_buffers=False, find_unused_parameters=True)
    elif (len(self.gpus) >= 2):
        self.parallel_model = DataParallel(self.model, self.gpus).to(self.device)
    else:
        self.parallel_model = self.model
def _parallelize(self):
    if self.multiprocess:
        self.model = convert_sync_bn(self.model).to(self.device)
        self.parallel_model = DistributedDataParallel(self.model, self.gpus, broadcast_buffers=False, find_unused_parameters=True)
    elif (len(self.gpus) >= 2):
        self.parallel_model = DataParallel(self.model, self.gpus).to(self.device)
    else:
        self.parallel_model = self.model