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