Neural Network from scratch in Python
- Python
- Numpy
- Scipy
- matplotlib.pyplot
- PIL
In main.py
a simple neural network with one convolutional layer and two fully connected layers.
All code can run without problem, but due to numerical unstability loss value might be inf. Please tell me if anyone has a solution.
To add layers to neural network and train it: python3 main.py
More details: 深度学习(五)上 Python徒手实现CNN