all_results.json

{
    "dr_accuracy": 0.99824,
    "dr_loss": 0.012910463847219944,
    "dr_runtime": 102.0036,
    "dr_samples_per_second": 245.089,
    "dr_steps_per_second": 0.961,
    "epoch": 20.0,
    "ood_accuracy": 0.8437441380604014,
    "ood_loss": 0.9615365266799927,
    "ood_runtime": 39.9719,
    "ood_samples_per_second": 266.737,
    "ood_steps_per_second": 1.051,
    "test_accuracy": 0.9126,
    "test_loss": 0.5343945026397705,
    "test_runtime": 101.9284,
    "test_samples_per_second": 245.27,
    "test_steps_per_second": 0.961,
    "train_loss": 0.08507246678442601,
    "train_runtime": 7726.1299,
    "train_samples_per_second": 64.715,
    "train_steps_per_second": 2.024
}