新力開源神經網路庫NNabla

發佈時間:2017-07-20 16:25:33 | 來源:選自Sony | 作者:佚名 | 責任編輯:胡俊

新力官方將很快發出更具體更詳細的神經網路庫使用方法。


新力表示他們的神經網路庫提供了最基本的深度學習開發研究和部署的函數。他們希望該軟體庫能推動人工智慧的發展,並且不論是在先進研究領域還是前沿應用部署上都能提供強有力的助力。

·         項目地址:https://github.com/sony/nnabla

·         説明文檔:https://nnabla.readthedocs.io/

·         項目官網:https://nnabla.org/

下面是該神經網路庫的Github項目介紹:


NNabla是一個深度學習框架,計劃在研究,開發與生産領域當中使用。新力的目標是讓它可以在任何地方運作,就像個人筆記型電腦、高性能計算集群、嵌入式裝置和生産伺服器一樣。因此新力希望通過提供高效便捷的機器學習庫而將演算法嵌入到邊緣設備中。


安裝

安裝NNabla很簡單:

pip install nnabla

這是 CPU 版本的NNabla,GPU-加速器可以通過安裝 CUDA 工具包來添加:pip install nnabla-ext-cuda


特徵

簡單,靈活並且有表現力

基於NNabla C++11 內核的 Python API 給你提供了靈活性和高産出效率。例如,一個帶有分類損失的兩層神經網路可以僅通過以下 5 行代碼被定義(超參數由<>括出)。

importnnablaasnn

importnnabla.functionsas F

importnnabla.parametric_functionsasPF

x =nn.Variable()

t =nn.Variable()

h =F.tanh(PF.affine(x,, name='affine1'))

y =PF.affine(h,, name='affine2')

loss=F.mean(F.softmax_cross_entropy(y, t))

1  2  3  4