iter_2/training_state.json

{
  "iter_num": 2,
  "local_iter_num": 2,
  "best_val_loss": 8.32339859008789,
  "running_mfu": -1.0,
  "lr": 1e-06,
  "epoch": 0,
  "step": 2
}