HebelGPU 加速深度學(xué)習(xí)庫

Hebel 是一款用 Python 編寫的支持 GPU 加速的深度學(xué)習(xí)庫。

Hebel 是一個通過 PyCUDA 庫使用 GPU CUDA 來加速建立神經(jīng)網(wǎng)絡(luò)的深度學(xué)習(xí)庫。它實現(xiàn)了幾類最重要的神經(jīng)網(wǎng)絡(luò)模型，提供各種激活函數(shù)和訓(xùn)練模型，包括 momentum、Nesterov momentum、dropout 和early stopping。

目前，Hebel  實現(xiàn)了前向式（feed-forward）神經(jīng)網(wǎng)絡(luò)模型，其他像自編碼（ Autoencoder）、卷積（Convolutional）神經(jīng)網(wǎng)絡(luò)模型和限制波爾茲曼機（Restricted Boltzman machines）等功能計劃后續(xù)加入。

Hebel 還實現(xiàn)了 L1 L2 權(quán)重衰減（weight decay）。

依賴

• PyCUDA

• numpy

• PyYAML

• skdata（僅用于 MNIST 示例）