整体架构

整个仓库

├── CHANGELOG.md
├── docs
├── examples
├── imgs
│   ├── logo2.png
│   └── logo.png
├── LICENSE
├── mkdocs.yml
├── pynet
├── README.EN.md
├── README.md
└── requirements.txt
  • PyNet实现代码位于pynet
  • PyNet示例代码位于examples
  • 说明文档位于docs

pynet

├── __init__.py
├── models
├── nn
├── optim
├── __pycache__
├── requirements.txt
├── solver.py
└── vision
  • 层的实现位于nn，比如Conv2d/MaxPool/Dropout/BN/ReLU/...
  • 模型的实现位于models，比如FCNet/LeNet5/AlexNet/NIN
  • 优化器和学习率调度器的实现位于optim，比如SGD/StepLR
  • 数据加载和训练结果绘制的实现位于vision
  • 求解器实现位于solver.py

examples

├── 2_nn_mnist.py
├── 3_nn_cifar10.py
├── 3_nn_iris.py
├── 3_nn_mnist.py
├── 3_nn_orl.py
├── lenet5_mnist.py
└── nin_cifar10.py

测试了4个网络模型：2层/3层神经网络、LeNet5和NIN；以及4个数据集：MNIST、CIFAR10、iris和ORL

Note：使用NIN的使用就很慢了