Spaces:
Sleeping
Sleeping
File size: 482 Bytes
9e4713f 743cb9b 4c71213 9e4713f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from pathlib import Path
from pydantic import BaseSettings
basedir = Path(__file__).parent
class Config(BaseSettings):
data_dir: Path = basedir.parent / "data"
raw_data_dir: Path = data_dir / "raw"
checkpoints_folders = data_dir / "checkpoints"
plots_dir: Path = data_dir / "plots"
batch_size: int = 64
epochs: int = 15
random_seed: int = 42
test_size: float = 0.2
weights_path: Path = data_dir / "models" / "model.torch"
config = Config()
|