1
0

use ignite for exp2

This commit is contained in:
2025-12-02 23:07:27 +08:00
parent 43b807679f
commit 65c56e938c
15 changed files with 246 additions and 794 deletions

12
exp2/modified/settings.py Normal file
View File

@@ -0,0 +1,12 @@
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"""