HyperParameters of this model

lr: 1e-5

epochs: 5

batch size: 1

grad acumulation step: 4