13 lines
304 B
Python
13 lines
304 B
Python
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'
|