summary.json

{
    "total_epochs": 150,
    "total_time_seconds": 23254.743891716003,
    "total_time_hours": 6.459651081032223,
    "batch_size_per_gpu": 2,
    "num_processes": 2,
    "mixed_precision": "fp16",
    "final_train_loss": 0.11351284384727478,
    "final_train_metrics": {
        "precision": [
            0.9284262657165527,
            0.9326722621917725,
            0.8191995024681091
        ],
        "recall": [
            0.9270820021629333,
            0.9292428493499756,
            0.8744466304779053
        ],
        "accuracy": [
            0.9996001720428467,
            0.9991011619567871,
            0.9927959442138672
        ],
        "dice": [
            0.9271973371505737,
            0.9303600788116455,
            0.8458120822906494
        ]
    },
    "final_val_metrics": {
        "precision": [
            0.8964494466781616,
            0.9273992776870728,
            0.7346808910369873
        ],
        "recall": [
            0.8001397848129272,
            0.8460437655448914,
            0.7222132682800293
        ],
        "accuracy": [
            0.9991061687469482,
            0.9987545013427734,
            0.8885079026222229
        ],
        "dice": [
            0.8246815204620361,
            0.8800930976867676,
            0.7170625329017639
        ]
    },
    "final_test_loss": 0.1920088231563568,
    "final_test_metrics": {
        "precision": [
            0.8555397391319275,
            0.93755042552948,
            0.7489745616912842
        ],
        "recall": [
            0.7621530294418335,
            0.838932991027832,
            0.7132463455200195
        ],
        "accuracy": [
            0.9988489151000977,
            0.9985865354537964,
            0.9637566804885864
        ],
        "dice": [
            0.7748545408248901,
            0.8739949464797974,
            0.7190028429031372
        ]
    }
}