all_results.json

{
    "dr_accuracy": 1.0,
    "dr_loss": 2.2834794435766526e-05,
    "dr_runtime": 30.7301,
    "dr_samples_per_second": 813.534,
    "dr_steps_per_second": 3.189,
    "epoch": 20.0,
    "ood_accuracy": 0.8552804351903958,
    "ood_loss": 1.3304697275161743,
    "ood_runtime": 13.3419,
    "ood_samples_per_second": 799.134,
    "ood_steps_per_second": 3.148,
    "test_accuracy": 0.90932,
    "test_loss": 0.8650785088539124,
    "test_runtime": 29.989,
    "test_samples_per_second": 833.64,
    "test_steps_per_second": 3.268,
    "train_loss": 0.0428766398688259,
    "train_runtime": 2610.1004,
    "train_samples_per_second": 191.564,
    "train_steps_per_second": 5.992
}