Removes from the of the parent of TFRobertaClassificationHead (#4884)

Co-authored-by: Hamza Harkous <harkous@google.com>
This commit is contained in:
Hamza Harkous
2020-06-09 16:14:01 -04:00
committed by GitHub
co-authored by Hamza Harkous
parent 02e5f79662
commit f0340b3031
+1 -1
View File
@@ -321,7 +321,7 @@ class TFRobertaClassificationHead(tf.keras.layers.Layer):
"""Head for sentence-level classification tasks."""
def __init__(self, config, **kwargs):
super().__init__(config, **kwargs)
super().__init__(**kwargs)
self.dense = tf.keras.layers.Dense(
config.hidden_size,
kernel_initializer=get_initializer(config.initializer_range),