checkpoints/training_log.json

{
  "step": [
    25,
    50,
    75,
    100
  ],
  "loss": [
    4.83566909790039,
    4.001754846572876,
    3.6740700721740724,
    3.5564898300170897
  ],
  "energy_initial": [
    1551801.665,
    1532687.17,
    1541912.385,
    1559144.77
  ],
  "energy_final": [
    605239.6925,
    603282.4875,
    602236.6225,
    604062.935
  ],
  "stationarity_residual": [
    1552.1239495849609,
    1549.458076171875,
    1548.9330493164061,
    1552.3543615722656
  ],
  "wall_time": [
    15.414806127548218,
    20.527431964874268,
    25.813805103302002,
    31.105968236923218
  ],
  "val_loss": [],
  "tokens_per_sec": [
    3321.4819295391003,
    4988.446688081726,
    5950.304474110719,
    6583.945512967493
  ]
}