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"""