{'loss': 1.3866, 'grad_norm': 0.25656482577323914, 'learning_rate': 4.708318749270797e-05, 'epoch': 0.18}
{'loss': 1.3856, 'grad_norm': 0.15997980535030365, 'learning_rate': 4.416637498541594e-05, 'epoch': 0.35}
{'loss': 1.3868, 'grad_norm': 0.17803697288036346, 'learning_rate': 4.124956247812391e-05, 'epoch': 0.53}
{'loss': 1.3868, 'grad_norm': 0.15967394411563873, 'learning_rate': 3.8332749970831875e-05, 'epoch': 0.7}
{'loss': 1.3866, 'grad_norm': 0.14670728147029877, 'learning_rate': 3.5415937463539846e-05, 'epoch': 0.88}