make exp3 works but no check
This commit is contained in:
@@ -1,14 +1,19 @@
|
||||
from pathlib import Path
|
||||
|
||||
POETRY_MAX_LEN: int = 64
|
||||
"""古诗词句子最大允许长度(该长度包含首尾填充的特殊字符),超过该长度的诗句将被删除。"""
|
||||
POETRY_MIN_WORD_FREQ: int = 8
|
||||
"""古诗词最小允许词频,小于该词频的词将在编解码时被视为[UNK]生僻字。"""
|
||||
|
||||
DIRTY_DATASET_PATH: Path = Path(__file__).resolve().parent.parent / 'datasets' / 'poetry.txt'
|
||||
"""脏的(未清洗的)古诗数据的路径"""
|
||||
CLEAN_DATASET_PATH: Path = Path(__file__).resolve().parent.parent / 'datasets' / 'poetry.pickle'
|
||||
"""干净的(已经清洗过的)古诗数据的路径"""
|
||||
|
||||
SAVED_MODULE_PATH: Path = Path(__file__).resolve().parent.parent / 'models' / 'rnn.pth'
|
||||
SAVED_MODEL_PATH: Path = Path(__file__).resolve().parent.parent / 'models' / 'rnn.pth'
|
||||
"""训练完毕的模型进行保存的路径"""
|
||||
|
||||
N_EPOCH: int = 10
|
||||
"""训练时的epoch"""
|
||||
N_BATCH_SIZE: int = 16
|
||||
N_BATCH_SIZE: int = 50
|
||||
"""训练时的batch size"""
|
||||
|
||||
Reference in New Issue
Block a user