GPU在Python中加速深度学习库
Hebel的Python项目详细描述
python中的gpu加速深度学习库
Hebel是python中使用神经网络进行深度学习的库 通过pycuda使用带有cuda的gpu加速。它实现了 重要类型的神经网络模型,并提供多种 不同的激活功能和训练方法,如动量, 内斯特罗夫动量,退出,并提前停止。
型号
现在,Hebel为 一个或多个任务的分类和回归。其他型号 例如自动编码器、卷积神经网络和受限玻尔兹曼 机器是为将来设计的。
Hebel为 正规化。
优化
Hebel用正则和 内斯特罗夫动量。
兼容性
目前,Hebel将在Linux和Windows上运行,可能还有MacOSX (未测试)。
依赖关系
- 皮库达
- 努比
- 皮亚姆
- skdata(仅适用于mnist示例)
安装
hebel在pypi上,因此您可以使用
pip install hebel
开始
研究examples/中的yaml配置文件并运行
python train_model.py examples/mnist_neural_net_shallow.yml
脚本将在examples/mnist中创建一个目录,其中 保存模型和日志。
阅读入门指南 hebel.readthedocs.org/en/latest/getting_started.html 更多信息。
文档
联系人
由Hannes Bretschneider维护 (hannes@psi.utoronto.ca)。如果你在用海贝尔,请告诉我 如果您发现任何错误,是否发现它有用并提交github问题 或者有功能要求。
名字是什么?
hebel是lever的德语单词,它是 人类使用。正如阿基米德所说:“给我一个足够长的杠杆和 它的支点,我将移动世界。”