Unverified Commit d2820758 authored by Sourav Singh's avatar Sourav Singh Committed by GitHub
Browse files

Remove cpu pin

parent 66628f7b

deepchem/models/tensorgraph/layers.py

+0 −1
Original line number Diff line number Diff line
@@ -2489,7 +2489,6 @@ class GraphGather(Layer): 
    super(GraphGather, self).__init__(**kwargs)



  def create_tensor(self, in_layers=None, set_tensors=True, **kwargs):

    with tf.device('/cpu'):

      inputs = self._get_input_tensors(in_layers)



      # x = [atom_features, deg_slice, membership, deg_adj_list placeholders...]

CRA Git | Maintained and supported by SUSTech CRA and CCSE