Hi @SShah30-hue

Can you post the requirements file for this repo? I am getting a lot of errors because of mismatches in python packages I believe. For example,

RoBERTa.ipynb:

from pytorch_lightning.metrics.functional import accuracy, f1, auroc

These metrics aren't supported by pytorch_lightning anymore (See this - Lightning-AI/lightning#2207 (comment))

I used torchmetrics and was able to proceed but got the following error later:

NotImplementedError: Support for training_epoch_end has been removed in v2.0.0. Tagger implements this method. You can use the on_train_epoch_end hook instead. To access outputs, save them in-memory as instance attributes. You can find migration examples in Lightning-AI/lightning#16520.

Thanks for your help!