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}