src/config.py

import os
import string
from dataclasses import dataclass

@dataclass
class Config:
    data_root: str = os.getenv("DATA_ROOT","Dataset_test\captchas")

    chars: str = string.ascii_letters + string.digits

    H: int = 48
    W_max: int = 224
    grayscale: bool = True  

    total_stride: int = 4  #
    batch_size: int = 32
    num_workers: int = 4
    amp: bool = True  

cfg = Config()