def set_weight_to_graph(self): self.graph.weighted = True for index, layer in enumerate(self.layers): set_keras_weight_to_stub(layer, self.graph.layer_list[index])