simpleT5 — Train T5 Models in Just 3 Lines of Code

Shivanand Roy
Geek Culture
Published in
3 min readJun 2, 2021

--

simpleT5 is built on top of PyTorch-lightning⚡️ and Transformers🤗 that lets you quickly train/fine-tune T5 models.

With simpleT5 — It is very easy to fine-tune any T5 model on your dataset (Pandas dataframe )— for any task (summarization, translation, question-answering, or other sequence-to-sequence tasks), just — import, instantiate, download a pre-trained model and train.

Checkout the simpleT5 GitHub repository (drop a 🌟 if you like it) along with example Colab notebook.

Before we jump on how to use simpleT5, a quick introduction about T5 —

What is T5 ?

A T5 is an encoder-decoder model. It converts all NLP problems like language translation, summarization, text generation, question-answering, to a text-to-text task.

For e.g., in case of translation from English →Serbian, T5 accepts source text: English, as input and tries to convert it into target text: Serbian.

--

--

Shivanand Roy
Geek Culture

Senior Data Scientist at Ernst & Young | Navigating the Future of AI and Tech | Sharing Unfiltered Reflections on Life’s Journey.