文件名称：etnn:Etnn是一个用C ++和CUDA编写的小型深度学习库
文件大小：196KB
文件格式：ZIP
更新时间：2024-05-10 15:01:32
C++
神经网络 Etnn是一个用C ++和CUDA编写的微型深度学习库。 您可以以此构建一个简单的分类器神经网络。 它基于cudnn和cublas构建，可为高效的操作员提供服务。 感谢提供的cudnn示例。 建造 cd etnn/ mkdir build && cd build/ cmake .. cmake --build . 例子 请首先将MNIST数据集下载到数据目录并运行示例： ./build/examples/mnist_mlp data/ 和分布式示例： ./local.sh 1 2 build/examples/mnist_mlp_distributed data
【文件预览】：
etnn-main
----CMakeLists.txt(744B)
----local.sh(630B)
----.gitignore(54B)
----README.md(545B)
----run_server.sh(384B)
----.gitmodules(85B)
----.clang-format(2KB)
----tests()
--------CMakeLists.txt(510B)
--------initial.cpp(1KB)
--------json.cpp(597B)
--------dataset.cpp(893B)
--------cublas.cpp(2KB)
--------softmax.cpp(2KB)
--------tensor.cpp(778B)
----examples()
--------CMakeLists.txt(321B)
--------mnist_mlp.cpp(1KB)
--------mnist_mlp_distributed.cpp(2KB)
--------json_mnist_mlp.cpp(1KB)
----include()
--------etnn()
--------nlohmann()
----src()
--------CMakeLists.txt(2KB)
--------dataset.cpp(3KB)
--------layer()
--------network.h(1KB)
--------loss()
--------metric()
--------utils.h(886B)
--------network_distibuted.cpp(8KB)
--------distributed()
--------tensor.h(4KB)
--------device.h(7KB)
--------network.cpp(5KB)
--------op()
--------dataset.h(762B)
--------parser.h(447B)
--------tensor.cpp(399B)
--------base.h(2KB)
--------network_distibuted.h(2KB)
--------initializer()
--------parser.cpp(4KB)
--------utils.cpp(3KB)
----ps-lite()
----json()
--------mnist.json(577B)
----kill.sh(73B)
----run_multi_worker.sh(441B)