SV12/ERA_Session13

1import toml2from pydantic import BaseModel3 4TOML_PATH = "config.toml"5 6 7class Data(BaseModel):8    batch_size: int = 5129    shuffle: bool = True10    num_workers: int = 411 12 13class LRFinder(BaseModel):14    numiter: int = 60015    endlr: float = 1016    startlr: float = 1e-217 18 19class Training(BaseModel):20    epochs: int = 2021    optimizer: str = "adam"22    criterion: str = "crossentropy"23    lr: float = 0.00324    weight_decay: float = 1e-425    lrfinder: LRFinder26 27 28class Config(BaseModel):29    data: Data30    training: Training31 32 33with open(TOML_PATH) as f:34    toml_config = toml.load(f)35 36config = Config(**toml_config)37