from transformers import BertModel

class MyBertModel(BertModel):
    def forward(self, *args, **kwargs):
        print("Success")
        return super().forward(*args, **kwargs)