Commit 01d03bfb authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

fix

parent 7eca4ad6

deepchem/rl/a2c.py

+0 −1
Original line number Diff line number Diff line
@@ -33,7 +33,6 @@ class A2CLossDiscrete(object): 
    self.value_index = value_index



  def __call__(self, outputs, labels, weights):

    import tensorflow as tf

    prob = outputs[self.action_prob_index]

    value = outputs[self.value_index]

    reward, advantage = weights

CRA Git | Maintained and supported by SUSTech CRA and CCSE