HyperParameters of this model

lr: 3e-5

epochs: 3

batch size: 1

grad acumulation step: 4