hyperparams:
lr = 5e-5 max_epochs = 20 batch size = 128
test set eval metrics:
{'eval_loss': 0.33326488733291626, 'eval_accuracy': 0.9, 'eval_precision': 0.9, 'eval_recall': 0.9, 'eval_f1': 0.9, 'eval_runtime': 3.5143, 'eval_samples_per_second': 5.691, 'eval_steps_per_second': 0.285}