LeNet5_FashionMNIST

train

mean=0.3814, std=0.3994
SGD weight_decay=0.1 추가
BatchNorm2d layer 추가
xavier 가중치 초기화

test

python main.py --mode test --output_dir ./output/ 
			   --checkpoint ./output/FASHION_MNIST/model_epoch3.pt

ref

Image Classification on Fashion-MNIST