from pathlib import Path

BATCH_SIZE: int = 16
"""训练的batch size"""

def get_saved_model_path() -> Path:
    """
    获取训练完毕的模型进行保存的路径。
    
    :return: 模型参数保存的路径。
    """
    return Path(__file__).resolve().parent.parent / 'models' / 'rnn.pth'