Loading deepchem/models/tensorgraph/tensor_graph.py +2 −0 Original line number Diff line number Diff line Loading @@ -601,6 +601,8 @@ class TensorGraph(Model): if not self.built: self.build() with self._get_tf("Graph").as_default(): if layer.variable_scope == '': return [] return tf.get_collection( tf.GraphKeys.TRAINABLE_VARIABLES, scope=layer.variable_scope) Loading deepchem/models/tensorgraph/tests/test_nbr_list.py +1 −0 Original line number Diff line number Diff line Loading @@ -74,6 +74,7 @@ class TestNbrList(test_util.TensorFlowTestCase): tg.set_loss(loss) tg.fit_generator(databag.iterbatches(epochs=1)) assert len(tg.get_layer_variables(combo)) >= 2 assert len(tg.get_layer_variables(out)) == 0 def test_neighbor_list_shape(self): """Test that NeighborList works.""" Loading Loading
deepchem/models/tensorgraph/tensor_graph.py +2 −0 Original line number Diff line number Diff line Loading @@ -601,6 +601,8 @@ class TensorGraph(Model): if not self.built: self.build() with self._get_tf("Graph").as_default(): if layer.variable_scope == '': return [] return tf.get_collection( tf.GraphKeys.TRAINABLE_VARIABLES, scope=layer.variable_scope) Loading
deepchem/models/tensorgraph/tests/test_nbr_list.py +1 −0 Original line number Diff line number Diff line Loading @@ -74,6 +74,7 @@ class TestNbrList(test_util.TensorFlowTestCase): tg.set_loss(loss) tg.fit_generator(databag.iterbatches(epochs=1)) assert len(tg.get_layer_variables(combo)) >= 2 assert len(tg.get_layer_variables(out)) == 0 def test_neighbor_list_shape(self): """Test that NeighborList works.""" Loading
