Unverified Commit f0340b30 authored by Hamza Harkous's avatar Hamza Harkous Committed by GitHub
Browse files

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


Co-authored-by: default avatarHamza Harkous <harkous@google.com>
parent 02e5f796
...@@ -321,7 +321,7 @@ class TFRobertaClassificationHead(tf.keras.layers.Layer): ...@@ -321,7 +321,7 @@ class TFRobertaClassificationHead(tf.keras.layers.Layer):
"""Head for sentence-level classification tasks.""" """Head for sentence-level classification tasks."""
def __init__(self, config, **kwargs): def __init__(self, config, **kwargs):
super().__init__(config, **kwargs) super().__init__(**kwargs)
self.dense = tf.keras.layers.Dense( self.dense = tf.keras.layers.Dense(
config.hidden_size, config.hidden_size,
kernel_initializer=get_initializer(config.initializer_range), kernel_initializer=get_initializer(config.initializer_range),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment