batch-size = 4
lr = 2.5e-5