DeepNeuralNets:具有GPU加速功能(OpenCL)的深度神经网络

时间:2021-05-01 11:25:37
【文件属性】:
文件名称:DeepNeuralNets:具有GPU加速功能(OpenCL)的深度神经网络
文件大小:86KB
文件格式:ZIP
更新时间:2021-05-01 11:25:37
Java 深度神经网络 用Java编写的具有GPU加速(OpenCL)的深度学习框架。 特征 可以与任何支持OpenCL(1.1)的GPU一起使用。 但是,仅在AMD和Apple平台中对代码进行过调整。 Nvidia和Intel尚未经过全面测试,但未预见到重大问题。 就像在Java中一样。 基本上没有操作系统限制。 支持卷积层,池化层和全连接层。 支持的激活类型包括Sigmoid,Tanh,Relu和Softmax。 训练使用具有可用参数(例如动量,重量衰减)的随机梯度下降法。 权重初始化具有高斯分布和均匀分布。 测试 该框架已使用MNIST和CIFAR-10(以Caffe为基准)进行了测试(PR2610 )。 使用相同的数据集,模型,参数,操作系统和硬件,此框架可在更短的训练时间内产生相同的分类精度。 对于MNIST,它的速度是CIFAR 2.6的2.4倍。
【文件预览】:
DeepNeuralNets-master
----.gitignore(390B)
----README.md(1007B)
----DeepNeuralNets()
--------.gitignore(13B)
--------src()
--------opencl()

网友评论