E_validate.py

"""
This script runs the nnU-Net Python API for model validation,
equivalent to the Command Line Interface (CLI) nnUNetv2_train command with --val flag.
"""
import torch

from custom_scripts.A_config import CONFIGURATION
from nnunetv2.run.run_training import run_training

if __name__ == '__main__':
    torch.set_num_threads(1)
    torch.set_num_interop_threads(1)
    CUDA = torch.device('cuda')
    CPU = torch.device('cpu')

    run_training(
        dataset_name_or_id="100",
        fold="2",
        device=CPU,
        configuration=CONFIGURATION,
        trainer_class_name='nnUNetTrainerFullOversamplingEarlyStopping',
        export_validation_probabilities=False,
        only_run_validation=True
    )