SHARKonvnet:GPU CNN支持SHARK

时间:2021-05-08 06:02:04
【文件属性】:
文件名称:SHARKonvnet:GPU CNN支持SHARK
文件大小:61KB
文件格式:ZIP
更新时间:2021-05-08 06:02:04
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)

网友评论