Skip to content

xiaoniute/CNN

Repository files navigation

Vectorized convolution neural network based on numpy

Environment requirement

cupy==4.1.0

matplotlib==3.0.0

numpy==1.15.2

python==3.5.6

scipy==1.1.0

tqdm==4.63.0、

Documentlist

train.py --- run this code to train the CNN

module.py --- document implementing the fodamental structure of CNN

cp_train.py --- cupy version

MNIST-Dataset --- folder storing the dataset

About

Vectorized convolution nerual network based on numpy

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages