开源中国

我们不支持 IE 10 及以下版本浏览器

It appears you’re using an unsupported browser

为了获得更好的浏览体验,我们强烈建议您使用较新版本的 Chrome、 Firefox、 Safari 等,或者升级到最新版本的IE浏览器。 如果您使用的是 IE 11 或以上版本,请关闭“兼容性视图”。
Sonnet首页、文档和下载 - 基于 TensorFlow 的神经网络库 - 开源中国社区
全部项目分类
我要评价
Apache
C/C++ Python
跨平台
Google
分享
收藏
85 人收藏
收录时间:2017-04-08
Sonnet 详细介绍

Sonnet 库使用面向对象的方法,允许创建定义一些前向传导计算的模块。模块用一些输入 Tensor 调用,添加操作到图里并返回输出 Tensor。其中一种设计选择是通过在随后调用相同的模块时自动重用变量来确保变量分享被透明化处理。

在各类文献中,很多模型都可以被视为分层形式,如可微分神经计算机可能包含 LSTM 控制器,可以实现为包含标准线性层。

该库兼容 Linux/Mac OS X 和 Python 2.7。TensorFlow 的版本必须至少为 1.0.1。Sonnet 支持 TensorFlow 的 virtualenv 安装模式,以及 nativ pip 安装。

安装步骤

若想安装 Sonnet,你需要使用 bazel 依靠 TensorFlow 头文件对这个库进行编译。请按照指定步骤安装 TensorFlow

安装 BAZEL

请确保你拥有最新版本的 bazel(至少为 0.4.5 版),如果版本过旧,请遵循此步骤

Virtualenv TensorFlow 安装

如果你想使用 virtualenv,请在安装时急活你的 virtualenv,或跳过此步骤:

$ source $VIRTUALENV_PATH/bin/activate # bash, sh, ksh, or zsh
$ source $VIRTUALENV_PATH/bin/activate.csh # csh or tcsh

配置 TensorFlow 头文件

复制 Sonnet 和 TensorFlow 的源代码作为一个子模块:

$ git clone --recursive https://github.com/deepmind/sonnet

使用 configure:

$ cd sonnet/tensorflow
$ ./configure
$ cd ../

这一步是让 Sonnet 构建于 TensorFlow 头文件上的必要步骤。

构建和运行安装程序

运行安装脚本,在临时目录中创建一个 wheel file:

$ mkdir /tmp/sonnet
$ bazel build --config=opt :install
$ ./bazel-bin/install /tmp/sonnet

pip install 生成的 wheel file:

$ pip install /tmp/sonnet/*.whl

如果已安装 Sonnet,卸载 wheel file 上之前的 pip install :

$ pip uninstall sonnet

通过尝试重采样(resampler op)来验证 Sonnet 是否安装成功:

$ cd ~/
$ python>>> import sonnet as snt>>> import tensorflow as tf>>> snt.resampler(tf.constant([0.]), tf.constant([0.]))

预计输出:

<tf.Tensor 'resampler/Resampler:0' shape=(1,) dtype=float32>

如果引入了 ImportError,C++组件未找到时,请确保没有导入复制的源代码(即在复制的资源库外调用 Python),并在安装 wheel file 前卸载 Sonnet。


大家对 Sonnet 的评论 (全部 0 条评论)
{{repayCom.userName}}
顶部