train_results.json

{
    "epoch": 6.885245901639344,
    "train_accuracy": 0.8871655680496316,
    "train_auc_class_0": 0.9601424899214485,
    "train_auc_class_1": 0.968345723853034,
    "train_auc_class_2": 0.9621813081120092,
    "train_auc_overall": 0.9635565072954971,
    "train_loss": 0.2833240032196045,
    "train_runtime": 718.0615,
    "train_samples_per_second": 10.775,
    "train_steps_per_second": 0.085
}