all_results.json

{
    "dr_accuracy": 0.9998,
    "dr_loss": 0.0014581895666196942,
    "dr_runtime": 108.3951,
    "dr_samples_per_second": 230.638,
    "dr_steps_per_second": 0.904,
    "epoch": 20.0,
    "ood_accuracy": 0.8211404989682987,
    "ood_loss": 1.3243283033370972,
    "ood_runtime": 46.2772,
    "ood_samples_per_second": 230.394,
    "ood_steps_per_second": 0.908,
    "test_accuracy": 0.89004,
    "test_loss": 0.8655158877372742,
    "test_runtime": 107.9984,
    "test_samples_per_second": 231.485,
    "test_steps_per_second": 0.907,
    "train_loss": 0.054148883717444245,
    "train_runtime": 8090.5095,
    "train_samples_per_second": 61.801,
    "train_steps_per_second": 1.933
}