Update modeling_norbert.py
Browse files- modeling_norbert.py +0 -4
modeling_norbert.py
CHANGED
|
@@ -387,7 +387,6 @@ class Classifier(nn.Module):
|
|
| 387 |
|
| 388 |
class NorbertForSequenceClassification(NorbertModel):
|
| 389 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 390 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 391 |
|
| 392 |
def __init__(self, config, **kwargs):
|
| 393 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -453,7 +452,6 @@ class NorbertForSequenceClassification(NorbertModel):
|
|
| 453 |
|
| 454 |
class NorbertForTokenClassification(NorbertModel):
|
| 455 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 456 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 457 |
|
| 458 |
def __init__(self, config, **kwargs):
|
| 459 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -501,7 +499,6 @@ class NorbertForTokenClassification(NorbertModel):
|
|
| 501 |
|
| 502 |
class NorbertForQuestionAnswering(NorbertModel):
|
| 503 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 504 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 505 |
|
| 506 |
def __init__(self, config, **kwargs):
|
| 507 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
@@ -569,7 +566,6 @@ class NorbertForQuestionAnswering(NorbertModel):
|
|
| 569 |
|
| 570 |
class NorbertForMultipleChoice(NorbertModel):
|
| 571 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
| 572 |
-
_keys_to_ignore_on_load_missing = ["head"]
|
| 573 |
|
| 574 |
def __init__(self, config, **kwargs):
|
| 575 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 387 |
|
| 388 |
class NorbertForSequenceClassification(NorbertModel):
|
| 389 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 390 |
|
| 391 |
def __init__(self, config, **kwargs):
|
| 392 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 452 |
|
| 453 |
class NorbertForTokenClassification(NorbertModel):
|
| 454 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 455 |
|
| 456 |
def __init__(self, config, **kwargs):
|
| 457 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 499 |
|
| 500 |
class NorbertForQuestionAnswering(NorbertModel):
|
| 501 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 502 |
|
| 503 |
def __init__(self, config, **kwargs):
|
| 504 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|
|
|
|
| 566 |
|
| 567 |
class NorbertForMultipleChoice(NorbertModel):
|
| 568 |
_keys_to_ignore_on_load_unexpected = ["classifier"]
|
|
|
|
| 569 |
|
| 570 |
def __init__(self, config, **kwargs):
|
| 571 |
super().__init__(config, add_mlm_layer=False, **kwargs)
|