all_results.json

{
    "dr_accuracy": 0.99856,
    "dr_loss": 0.0108617153018713,
    "dr_runtime": 30.0882,
    "dr_samples_per_second": 830.891,
    "dr_steps_per_second": 3.257,
    "epoch": 20.0,
    "ood_accuracy": 0.8312699305946352,
    "ood_loss": 1.032335638999939,
    "ood_runtime": 12.2571,
    "ood_samples_per_second": 869.86,
    "ood_steps_per_second": 3.427,
    "test_accuracy": 0.90164,
    "test_loss": 0.5982226729393005,
    "test_runtime": 29.375,
    "test_samples_per_second": 851.064,
    "test_steps_per_second": 3.336,
    "train_loss": 0.0671754312637212,
    "train_runtime": 2479.9785,
    "train_samples_per_second": 201.615,
    "train_steps_per_second": 6.307
}