This repo provides the example of LeNet-5 Model implemented by PyTorch.
LeNet-4 is the one of most popular models for the numerical recognition.
It is consist of the feature extraction(CNN) and the classification(FC).
- Python3
- PyTorch
- Install in
Conda
environment if necessary
- Install in
- (option) conda active
- model train
- model test
conda active {my_env}
python LeNet5_train.py
python LeNet5_test.py