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):
"""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),
......
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