77143fb 1ddc92c
1
2
3
4
5
6
7
from transformers import BertModel class MyBertModel(BertModel): def forward(self, *args, **kwargs): print("Success") return super().forward(*args, **kwargs)
