Commit 586871a1 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #847 from peastman/fetches

Minor optimization to TensorGraph
parents 20edb37c 56fe2be5
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -189,8 +189,7 @@ class TensorGraph(Model):
            args=(self, feed_dict_generator, self._get_tf("Graph"),
                  self.session, coord))
        enqueue_thread.start()
      output_tensors = [x.out_tensor for x in self.outputs]
      fetches = output_tensors + [train_op, self.loss.out_tensor]
      fetches = [train_op, self.loss.out_tensor]
      for feed_dict in create_feed_dict():
        try:
          fetched_values = self.session.run(fetches, feed_dict=feed_dict)