/transformers

This repository contains code to train, fine tune & use different transformer models

Primary LanguageJupyter Notebook

  • Code to generate embeddings using BERT forward pass Here
  • Code to finetune BERT using Trainer Here
  • Code to finetune BERT using Native pytorch Here