此项目用于 ECNU 计算机科学与技术系优化理论及方法课大作业。
目标为实现一个简易的神经网络结构,并能运行一些简单的神经网络训练项目。
待完成目标:
- Tensor 类
- Layer 类:
- 卷积层
- 全连接层
- 最大池化层
- 平均池化层
- 拉伸层
- Optimizer 类:
- SGD
- Activation 类:
- Sigmoid 函数
- ReLU 函数
- loss 类:
- 交叉熵损失函数
- Model 类:
- 简单顺序连接模型
此项目用于 ECNU 计算机科学与技术系优化理论及方法课大作业。
目标为实现一个简易的神经网络结构,并能运行一些简单的神经网络训练项目。
待完成目标: