Skip to content

heyxhh/nnet-numpy

Repository files navigation

神经网络numpy实现

介绍

  • Linear、CNN、Maxpool、BatchNorm、Dropout、SoftmaxLoss等,正向反向传播的实现。
  • 代码包含大量的中文注释,对于刚接触深度学习的同学可以作为参考,用来当玩具练习。
  • 代码测试有效,设计了类似AlexNet的简单卷积网络,使用mnist数据集训练3个epoch后test accuracy(98%)。
  • 实验效果查看net-test-mnist.ipynb

About

CNN、Linear网络纯numpy实现

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages