The idea is that i\'m building a custom model (class that implements nn.Module) using pytorch this model form huggingface transformers to fine-tune pretrained bert: