model/config.py

class UNetConfig:
    def __init__(self):
        # Configuraciones del modelo
        self.in_channels = 3  # Por ejemplo, para imágenes RGB
        self.out_channels = 21  # Supongamos una tarea de segmentación binaria
        self.encoder_name = "resnet34"
        self.encoder_weights = "imagenet"

        # Configuraciones del DataLoader
        self.batch_size = 32
        self.num_workers = 4  # Dependiendo de tu máquina
        self.shuffle = True

        # Otros parámetros de entrenamiento podrían ir aquí
        self.learning_rate = 1e-3