Commit 6ceecc9b authored by peastman's avatar peastman
Browse files

Bug fix

parent 2388e843

deepchem/models/tensorgraph/tensor_graph.py

+2 −2
Original line number Diff line number Diff line
@@ -415,9 +415,9 @@ class TensorGraph(Model): 
      sorted_layers.append(layer)



    sorted_layers = []

    for l in self.features + self.labels + self.task_weights + self.outputs:

      add_layers_to_list(l, sorted_layers)

    add_layers_to_list(self.loss, sorted_layers)

    for o in self.outputs:

      add_layers_to_list(o, sorted_layers)

    return sorted_layers



  def build(self):

CRA Git | Maintained and supported by SUSTech CRA and CCSE