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

fix

parent 7eca4ad6
......@@ -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
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment