{'loss': 1.3867, 'grad_norm': 0.192262664437294, 'learning_rate': 4.708318749270797e-05, 'epoch': 0.18}
{'loss': 1.3869, 'grad_norm': 0.16380712389945984, 'learning_rate': 4.416637498541594e-05, 'epoch': 0.35}
{'loss': 1.3869, 'grad_norm': 0.16317473351955414, 'learning_rate': 4.124956247812391e-05, 'epoch': 0.53}
{'loss': 1.3868, 'grad_norm': 0.1727585792541504, 'learning_rate': 3.8332749970831875e-05, 'epoch': 0.7}
{'loss': 1.3866, 'grad_norm': 0.15967760980129242, 'learning_rate': 3.5415937463539846e-05, 'epoch': 0.88}