Python get_network示例

示例#1

 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()

示例#2

 def __create_network(self):
     self.net = get_network(self.config['network'])
     self.net = nn.DataParallel(self.net, device_ids=[0, 1])
     self.net.double()

示例#3

 def __create_network(self):
     self.net = get_network(self.config['network'])
     self.net.double()