I implemented the training loop in Pytorch Lightning which can be found in the file train_lightning.py.
praeclarumjj3 / pytorch-lightning-assignment Goto Github PK
View Code? Open in Web Editor NEWI had to write the training loop for training BERT using PyTorch Lightning [at OffNote Labs]