use ignite for exp2
This commit is contained in:
12
exp2/modified/settings.py
Normal file
12
exp2/modified/settings.py
Normal 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"""
|
||||
Reference in New Issue
Block a user