utils.py

# utils.py
import random
import numpy as np
import torch
import os

def set_seeds(seed_value):
    """Sets random seeds for reproducibility."""
    random.seed(seed_value)
    np.random.seed(seed_value)
    torch.manual_seed(seed_value)
    if torch.cuda.is_available():
        torch.cuda.manual_seed(seed_value)
        torch.cuda.manual_seed_all(seed_value)
        torch.backends.cudnn.deterministic = True
        torch.backends.cudnn.benchmark = False
    print(f"Seeds set to {seed_value}")

def ensure_dir(directory_path):
    """Ensures that a directory exists, creating it if necessary."""
    if not os.path.exists(directory_path):
        try:
            os.makedirs(directory_path)
            print(f"Directory created: '{directory_path}'")
        except OSError as e:
            print(f"Error creating directory '{directory_path}': {e}")
    # else:
    #     print(f"Directory already exists: '{directory_path}'")