Merge 1aaa718 into 138522e

deepchem · Jul 21, 2020 · 3ac85c9 · 3ac85c9
2 parents 138522e + 1aaa718
commit 3ac85c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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:
@@ -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()