Exemplo n.º 1
0
 def set_weight_to_graph(self):
     self.graph.weighted = True
     for index, layer in enumerate(self.layers):
         set_torch_weight_to_stub(layer, self.graph.layer_list[index])
Exemplo n.º 2
0
 def set_weight_to_graph(self):
     self.graph.weighted = True
     for index, layer in enumerate(self.layers):
         set_torch_weight_to_stub(layer, self.graph.layer_list[index])