Commit 3162ce78 authored by peastman's avatar peastman
Browse files

Bug fix to converting layers to tensors

parent 24437fc5

deepchem/models/tensorgraph/layers.py

+3 −0
Original line number Diff line number Diff line
@@ -162,7 +162,10 @@ class Layer(object): 
      tf.summary.histogram(self.name, self.tb_input, self.collections)



  def _as_graph_element(self):

    if '_as_graph_element' in dir(self.out_tensor):

      return self.out_tensor._as_graph_element()

    else:

      return self.out_tensor





def _convert_layer_to_tensor(value, dtype=None, name=None, as_ref=False):

CRA Git | Maintained and supported by SUSTech CRA and CCSE