src/config/config.py

from dataclasses import dataclass


@dataclass
class ModelConfig:
    bert_output_size = 312
    embedding_size = 128


@dataclass
class TrainConfig:
    epochs = 12
    batch_size = 16