checkpoint_info.json

{
  "epoch": 2,
  "metrics": {
    "loss/MI": 1.0626029676447313,
    "loss/TA": 1.2027087981502216,
    "loss/total": 2.2653117713828883,
    "train/loss_avg": 2.2653117713828883,
    "val/loss": 2.3157517766952513,
    "val/MI_pcc": 0.8138228058815002,
    "val/MI_srcc": 0.8251110190549646,
    "val/TA_pcc": 0.5718650817871094,
    "val/TA_srcc": 0.5726515902330593,
    "epoch": 2,
    "time_sec": 54.221648931503296,
    "lr": 6.183999999999998e-05
  }
}