praeclarumjj3 / Pytorch-Lightning-Assignment

I had to write the training loop for training BERT using PyTorch Lightning [at OffNote Labs]

Pytorch Lightning Assignment

I implemented the training loop in Pytorch Lightning which can be found in the file train_lightning.py.