13 lines
385 B
Python
13 lines
385 B
Python
|
from pathlib import Path
|
||
|
|
|
||
|
|
MNIST_DATASET_PATH: Path = Path(__file__).resolve().parent.parent / 'datasets' / 'mnist.npz'
|
||
|
|
"""MNIST数据集文件的路径"""
|
||
|
|
|
||
|
|
SAVED_MODEL_PATH: Path = Path(__file__).resolve().parent.parent / 'models' / 'cnn.pth'
|
||
|
|
"""训练好的模型保存的位置"""
|
||
|
|
|
||
|
|
N_EPOCH: int = 5
|
||
|
|
"""训练时的epoch次数"""
|
||
|
|
N_BATCH_SIZE: int = 1000
|
||
|
|
"""训练时的batch size"""
|