文件名称：SHARKonvnet:GPU CNN支持SHARK
文件大小：61KB
文件格式：ZIP
更新时间：2024-05-27 23:48:44
C++
鲨鱼网络 GPU CNN适用于SHARK。 安装 Cuda工具包 cuDNN 鲨鱼 生成文件 以下是有关如何进行编译的提示。 MNIST 将目录更改为scripts然后运行python prep_mnist_data.py 。 这将下载，解压缩MNIST数据集并将其标准化为data/mnist/ 。 将目录更改为项目的根目录，然后运行make mnist 。 这将把tests/MnistTest.cpp编译成obj/MnistTest.o 。 使用./obj/MnistTest.o运行。 您应该看到有关convnet的一些信息（在tests/MnistTest.cpp指定），然后每隔500次迭代就会看到一些进度信息。 经过2000次迭代后，01损耗应为4％左右。 它收敛到略低于1％。
【文件预览】：
SHARKonvnet-master
----src()
--------layers()
--------ConvNet.h(14KB)
----scripts()
--------prep_mnist_data.py(2KB)
----LICENSE(7KB)
----README.md(787B)
----makefile(2KB)
----data()
--------mnist()
--------.gitignore(0B)
----tests()
--------CifarTest.cpp(11KB)
--------DistortLayerToDiskTest.cpp(4KB)
--------MnistTest.cpp(11KB)
--------LayersGradientUnitTest.cpp(23KB)
--------DistortTest.cu(9KB)
--------ConvNetTests.cpp(13KB)
--------derivativeTestHelper.h(14KB)