Skip to content

v0.3.1

Compare
Choose a tag to compare
@gagan3012 gagan3012 released this 15 Jun 21:23
· 958 commits to master since this release

This version features a tested trainer which can be used in 4 lines of code:

from keytotext import KeytotextTrainer

model = KeytotextTrainer()
model.from_pretrained(model_name="t5-small")
model.train(data_df=df,batch_size=4, max_epochs=3, use_gpu=True)
model.save_model()