| 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 | |