文件名称:CUDA-DNN-MNIST:在MNIST数据集上训练神经网络的示例C ++ CUDA实现
文件大小:1.7MB
文件格式:ZIP
更新时间:2024-06-01 22:04:47
cpp cuda mnist-dataset JupyterNotebook
CUDA DNN MNIST 该项目是在纯C ++ CUDA代码上的MNIST数据集上训练简单前馈神经网络的示例实现。 注意:该项目仍在开发中,仅是出于娱乐目的而创建,并在我的大学通过CUDA项目。 除实验外,请勿将其用于其他任何用途-我无法保证它会起作用! :) 随意分叉,实施和试验! 怎么跑? 要编译和运行该项目,您只需要Make和NVCC。 然后,只需运行: $ > make dataset $ > make build $ > make run 输出示例: ===================================== Configuration ===================================== NumberOfEpochs: 100 BatchSize: 512 LearningRate: 1.000
【文件预览】:
CUDA-DNN-MNIST-master
----.gitignore(98B)
----README.md(19KB)
----LICENSE(34KB)
----src()
--------configuration.cuh(2KB)
--------loggers()
--------loss()
--------datasets()
--------utils.hpp(516B)
--------models()
--------layers()
--------optimizers()
--------main.cu(5KB)
--------tensor()
--------utils.cpp(220B)
--------configuration.cu(4KB)
----experiments()
--------.gitignore(93B)
--------Experiments Analysis.ipynb(1.03MB)
--------experiments.py(7KB)
--------run.py(1KB)
--------logs.zip(623KB)
----Makefile(1KB)
----assets()
--------experiment2_small_backward.png(17KB)
--------experiment1_small_backward_first.png(18KB)
--------experiment6_small_forward.png(16KB)
--------experiment6_medium_forward.png(16KB)
--------experiment5_medium_backward.png(23KB)
--------experiment2_large_forward.png(17KB)
--------experiment2_small_forward.png(17KB)
--------experiment6_medium_backward.png(17KB)
--------experiment2_large_backward.png(17KB)
--------experiment4_medium_backward.png(17KB)
--------experiment6_large_backward.png(17KB)
--------experiment1_medium_forward_first.png(18KB)
--------experiment3_large_forward.png(18KB)
--------experiment3_large_backward.png(18KB)
--------experiment5_medium_forward.png(21KB)
--------experiment1_large_backward_first.png(18KB)
--------experiment1_large_backward_second.png(17KB)
--------experiment4_medium_forward.png(17KB)
--------experiment5_small_forward.png(22KB)
--------experiment1_large_forward_second.png(17KB)
--------experiment6_large_forward.png(17KB)
--------experiment3_medium_forward.png(18KB)
--------experiment2_medium_forward.png(17KB)
--------experiment1_small_forward_first.png(18KB)
--------experiment1_medium_backward_second.png(17KB)
--------experiment3_small_backward.png(17KB)
--------experiment1_small_forward_second.png(17KB)
--------experiment4_large_forward.png(17KB)
--------experiment2_medium_backward.png(17KB)
--------experiment5_large_backward.png(23KB)
--------experiment5_large_forward.png(22KB)
--------experiment1_large_forward_first.png(18KB)
--------experiment3_small_forward.png(17KB)
--------experiment4_small_forward.png(17KB)
--------experiment4_large_backward.png(17KB)
--------experiment1_small_backward_second.png(17KB)
--------experiment5_small_backward.png(22KB)
--------experiment1_medium_forward_second.png(18KB)
--------experiment4_small_backward.png(17KB)
--------experiment1_medium_backward_first.png(17KB)
--------experiment3_medium_backward.png(18KB)
--------experiment6_small_backward.png(18KB)