Unverified Commit f4fb1f00 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #948 from lilleswing/tensorboard 

Write Values to Tensorboard not Tensors
parents b9b35840 e2449b07

deepchem/models/tensorgraph/tensor_graph.py

+1 −1
Original line number Diff line number Diff line
@@ -232,7 +232,7 @@ class TensorGraph(Model): 
          fetches.append(self._get_tf("summary_op"))

        fetched_values = self.session.run(fetches, feed_dict=feed_dict)

        if should_log:

          self._log_tensorboard(fetches[2])

          self._log_tensorboard(fetched_values[2])

        avg_loss += fetched_values[1]

        n_averaged_batches += 1

        self.global_step += 1

deepchem/models/tensorgraph/tests/test_tensor_graph.py

+1 −1
Original line number Diff line number Diff line
@@ -191,7 +191,6 @@ class TestTensorGraph(unittest.TestCase): 
    prediction2 = np.squeeze(tg1.predict_on_batch(X))

    assert_true(np.all(np.isclose(prediction, prediction2, atol=0.01)))



  @nottest

  def test_tensorboard(self):

    n_data_points = 20

    n_features = 2
@@ -208,6 +207,7 @@ class TestTensorGraph(unittest.TestCase): 
        tensorboard=True,

        tensorboard_log_frequency=1,

        learning_rate=0.01,

        use_queue=False,

        model_dir='/tmp/tensorgraph')

    tg.add_output(output)

    tg.set_loss(loss)

CRA Git | Maintained and supported by SUSTech CRA and CCSE