具有并行计算的卷积神经网络的C ++库(openMP,CUDA,MPI)
用法:
g ++ -std = c ++ 11 -fopenmp lenet.cpp -o lenet ./lenet
这是模型的多线程版本(具有数据并行性),您可以使用以下方法更改线程数: 导出OMP_NUM_THREADS = 4
要使用MPI版本的代码,您需要使用mpic ++进行编译: mpic ++ -std = c ++ 11 -fopenmp lenet.cpp -o lenet
您可以在多节点系统上运行它!
创建自己的网络
您可以通过派生Model类并使用addLayer()方法按顺序添加所有图层来创建自己的深度神经网络类。
您还可以通过扩展ActivationLayer来引入自己的激活层。
您可以通过扩展LossFunction类来创建自定义Loss函数。
工作正在进行中
使用以下方法进
2022-05-13 18:00:15
10.98MB
C++
1