{
    "epoch": 3.0,
    "train_accuracy": 0.13333333333333333,
    "train_auc_class_0": 0.39999999999999997,
    "train_auc_class_1": 0.6000000000000001,
    "train_auc_class_2": 0.2,
    "train_auc_overall": 0.39999999999999997,
    "train_loss": 1.3039982318878174,
    "train_runtime": 1.7052,
    "train_samples_per_second": 8.797,
    "train_steps_per_second": 0.586
}