LeNet

To train the model please run train.py

And test the model by test.py

Or you can directly run test.py to use my trained model.pth to make a classification

It is a very simple model and it's my homework. Good luck to everyone!