update for change of exp2 and add exp3
This commit is contained in:
12
exp3/modified/settings.py
Normal file
12
exp3/modified/settings.py
Normal file
@@ -0,0 +1,12 @@
|
||||
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'
|
||||
Reference in New Issue
Block a user