File size: 228 Bytes
ef18bd7 |
1 2 3 4 5 6 7 8 |
from transformers import BertForSequenceClassification
class PatchedBertForSequenceClassification(BertForSequenceClassification):
def __reduce__(self):
print("hello!")
return self.__class__, (self.config,)
|