Training General Pytorch model with HuggingFace's Trainer

🀗Transformers
1

Hello,

I was wondering if we could utilize HuggingFace’s Trainer API to train the PyTorch model. In PyTorch Lightning, we can conveniently adapt our existing PyTorch model by inheriting the PyTorch model with Pt Lightning Module regardless of the model architecture. If this is possible, could someone kindly recommend a helpful resource?

Thanks in advance,
Ransaka

1 Like