-
Notifications
You must be signed in to change notification settings - Fork 4
TomFrank0217/cnn
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
这个小工程用C++从最底层的矩阵运算，到三维四维的张量运算，复现了lenet的正向和反向传播的细节。 经过调参，在 lenet的测试集上达到了99.03%的准确度。 But, there are some problems todo. 1. 补充CMakeLists方便在linux下调试 2. 代码量4000行太多了，不易于维护，重构封装函数，增加注释，减少代码量，提高代码的可读性 3. layers_parameters.cpp 中是描述网络结构的参数，现在不支持跨层的网络结构 4. 修改网络正向传播和反向传播的细节，减少重复的运算量 5. 修改损失函数，使得准确度提升 6. 想支持shufflenet与resnet的网络结构。 7. 将读取图像和前向传播设计成生产者消费者模式，加快训练速度。
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Packages 0
No packages published