Skip to content

Commit

Permalink
Merge pull request #1987 from chstem/fix
Browse files Browse the repository at this point in the history
fix in GraphConvModel
  • Loading branch information
Bharath Ramsundar committed Jul 8, 2020
2 parents 01e37ea + 0275c97 commit 5202bd7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepchem/models/graph_models.py
Expand Up @@ -635,7 +635,7 @@ def call(self, inputs, training=False):
if self.batch_norms[-1] is not None:
dense = self.batch_norms[-1](dense, training=training)
if training and self.dropouts[-1] is not None:
dense = self.dropouts[1](dense, training=training)
dense = self.dropouts[-1](dense, training=training)
neural_fingerprint = self.graph_gather([dense, degree_slice, membership] +
deg_adjs)
if self.mode == 'classification':
Expand Down

0 comments on commit 5202bd7

Please sign in to comment.