metrics_epoch_1.json

{
  "best_epoch": null,
  "peak_worker_0_memory_MB": 4512.1484375,
  "peak_gpu_0_memory_MB": 2972.189453125,
  "training_duration": "0:05:46.709385",
  "epoch": 1,
  "training_loss": 1.6510879126760434,
  "training_worker_0_memory_MB": 4512.1484375,
  "training_gpu_0_memory_MB": 2972.189453125
}