Ejemplo n.º 1
0
 def create_network(self):
     if (self.net is None):
         self.net = get_network(self.config['network'])
     if (self.tensor_type == 'float'):
         self.net.float()
     else:
         self.net.double()
Ejemplo n.º 2
0
 def __create_network(self):
     self.net = get_network(self.config['network'])
     self.net = nn.DataParallel(self.net, device_ids=[0, 1])
     self.net.double()
Ejemplo n.º 3
0
 def __create_network(self):
     self.net = get_network(self.config['network'])
     self.net.double()