{
    "epoch": 3.0,
    "train_accuracy": 0.7333333333333333,
    "train_auc_class_0": 0.8400000000000001,
    "train_auc_class_1": 0.9199999999999999,
    "train_auc_class_2": 1.0,
    "train_auc_overall": 0.9199999999999999,
    "train_loss": 0.7121346592903137,
    "train_runtime": 1.6755,
    "train_samples_per_second": 8.952,
    "train_steps_per_second": 0.597
}