Ejemplos de convert_sync_bn en Python

Ejemplo n.º 1

Archivo: ofa.py Proyecto: zeta1999/aw_nas

 def _parallelize(self):
     if self.multiprocess:
         net = convert_sync_bn(self).to(self.device)
         object.__setattr__(
             self, "parallel_model",
             DistributedDataParallel(net,
                                     self.gpus,
                                     find_unused_parameters=True))

Ejemplo n.º 2

Archivo: cnn_trainer.py Proyecto: ZhuangzhuangWu/aw_nas

 def _parallelize(self):
     if self.multiprocess:
         net = convert_sync_bn(self.model).to(self.device)
         self.parallel_model = DistributedDataParallel(
             net, self.gpus, 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