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

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

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

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

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

依賴

• PyCUDA

• numpy

• PyYAML

• skdata（僅用于 MNIST 示例）