npnn

NumPy Neural Network

What's npnn?

npnn is a a torch-like Python module for gradient descent based machine learning implemented with numpy.

Dependency

Basically npnn only depends on numpy(the latest version 1.26.4 is verified).

If you have CUDA devices available, then you can easily get a acceleration by installing suitable version of cupy. In this case npnn will use cupy api rather than numpy api.

For example, my PC have CUDA v12.x (x86_64), so I use command:

pip install cupy-cuda12x
pip install npnn

or in short:

pip install npnn[cuda12x]

check cupy documentation for more information.

API references

See npnn WIKI.

Known issues

See npnn known-issues.

Work with npnn!

Here we will construct a image classification neural network with npnn.

BTW, this is a course assignment of DATA620004, School of Data Science, Fudan University.

Task

Construct and Train a neural network on Fashion-MNIST to do image classification.

• Implement gradient backpropagation algorithm by hand,you can use numpy but DO NOT use pytorch or tensorflow to do autograd.

• Submit source code including at least four parts: model definition, training, parameters searching and testing.

Implementation

• dataset.py: provide Fashion MNIST dataset
• model.py: model definition
• train.py: model training
• search.py: parameters searching
• test.py: model testing
• viz.py: visualization
• utils.py: some misc function, such as save_model

run search.py, you can get a table like:

train log file and saved model weights can be found in ./logs and ./checkpoints folder.