File size: 291 Bytes
54f5ec9 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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('.')
|