From a60bc6856117f104bc7b6755cab4e3a65fe21124 Mon Sep 17 00:00:00 2001 From: Bharath Ramsundar Date: Mon, 20 Jul 2020 18:33:26 -0700 Subject: [PATCH 1/2] Changes --- deepchem/models/graph_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepchem/models/graph_models.py b/deepchem/models/graph_models.py index 51006352a5..5a5599c1d2 100644 --- a/deepchem/models/graph_models.py +++ b/deepchem/models/graph_models.py @@ -265,7 +265,7 @@ def __init__(self, output_types = ['prediction', 'loss'] loss: Loss = SoftmaxCrossEntropy() else: - output = Dense(n_tasks)(weave_gather) + output = Dense(n_tasks)(output) outputs = [output] output_types = ['prediction'] loss = L2Loss() From 1aaa7180ed9ebe6bda01140ffc8f3ee54d6b6889 Mon Sep 17 00:00:00 2001 From: Bharath Ramsundar Date: Mon, 20 Jul 2020 18:35:12 -0700 Subject: [PATCH 2/2] Fix --- deepchem/models/graph_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepchem/models/graph_models.py b/deepchem/models/graph_models.py index 5a5599c1d2..0fbcb3deae 100644 --- a/deepchem/models/graph_models.py +++ b/deepchem/models/graph_models.py @@ -244,7 +244,7 @@ def __init__(self, kernel_initializer=tf.keras.initializers.TruncatedNormal( stddev=weight_stddev), bias_initializer=tf.constant_initializer(value=bias_const), - kernel_regularizer=regularizer)(weave_gather) + kernel_regularizer=regularizer)(input_layer) if dropout > 0.0: layer = Dropout(rate=dropout)(layer) if batch_normalize: