fix: fix typo
Browse files- modeling_bert.py +1 -1
modeling_bert.py
CHANGED
@@ -298,7 +298,7 @@ class BertPreTrainingHeads(nn.Module):
|
|
298 |
return prediction_scores, seq_relationship_score
|
299 |
|
300 |
|
301 |
-
class BertPreTrainedModel(
|
302 |
"""An abstract class to handle weights initialization and
|
303 |
a simple interface for dowloading and loading pretrained models.
|
304 |
"""
|
|
|
298 |
return prediction_scores, seq_relationship_score
|
299 |
|
300 |
|
301 |
+
class BertPreTrainedModel(PreTrainedModel): #(nn.Module):
|
302 |
"""An abstract class to handle weights initialization and
|
303 |
a simple interface for dowloading and loading pretrained models.
|
304 |
"""
|