| from happytransformer import HappyTextToText, TTTrainArgs | |
| happy_tt = HappyTextToText("T5", load_path=".") | |
| args = TTTrainArgs(batch_size=8) | |
| happy_tt.train("data/train.csv", args=args) | |
| before_loss = happy_tt.eval("data/eval.csv") | |
| print("After loss: ", before_loss.loss) | |
| happy_tt.save('.') | |