{% extends "!layout.html" %} {% set title = "欢迎使用 Neural Network Intelligence !!!"%} {% block document %}
English
NNI (Neural Network Intelligence) 是一个轻量但强大的工具包,帮助用户自动的进行 特征工程神经网络架构搜索超参调优以及模型压缩

NNI 管理自动机器学习 (AutoML) 的 Experiment, 调度运行 由调优算法生成的 Trial 任务来找到最好的神经网络架构和/或超参,支持 各种训练环境,如 本机, 远程服务器, OpenPAI, Kubeflow, 基于 K8S 的 FrameworkController(如,AKS 等) DLWorkspace (又称 DLTS),以及其它云服务。

使用场景

NNI {{ release }} 已发布!

NNI 功能一览

NNI 提供命令行工具以及友好的 WebUI 来管理训练的 Experiment。 通过可扩展的 API,可定制自动机器学习算法和训练平台。 为了方便新用户,NNI 内置了最新的自动机器学习算法,并为流行的训练平台提供了开箱即用的支持。

下表中,包含了 NNI 的功能,同时在不断地增添新功能,也非常希望您能贡献其中。

框架和库 算法 训练平台
内置
  • 支持的框架
    • PyTorch
    • Keras
    • TensorFlow
    • MXNet
    • Caffe2
    • 更多...
  • 支持的库
超参调优 神经网络架构搜索 模型压缩 特征工程(测试版) 提前终止算法
参考

安装

安装

NNI 支持并在 Ubuntu >= 16.04, macOS >= 10.14.1, 和 Windows 10 >= 1809 通过了测试。 在 `python 64 位 >= 3.5` 的环境中,只需要运行 `pip install` 即可完成安装。

Linux 或 macOS
python3 -m pip install --upgrade nni
Windows
python -m pip install --upgrade nni

如果想要尝试最新代码,可通过源代码安装 NNI

Linux 和 macOS 下 NNI 系统需求参考这里,Windows 参考这里

注意:

验证安装

以下示例基于 TensorFlow 1.x 构建。 确保运行环境中使用的是 TensorFlow 1.x

文档

贡献

本项目欢迎任何贡献和建议。 大多数贡献都需要你同意参与者许可协议(CLA),来声明你有权,并实际上授予我们有权使用你的贡献。 有关详细信息,请访问 https://cla.microsoft.com

当你提交拉取请求时,CLA 机器人会自动检查你是否需要提供 CLA,并修饰这个拉取请求(例如,标签、注释)。 只需要按照机器人提供的说明进行操作即可。 CLA 只需要同意一次,就能应用到所有的代码仓库上。

该项目采用了 Microsoft 开源行为准则 。 有关详细信息,请参阅行为守则常见问题解答或联系 opencode@microsoft.com 咨询问题或评论。

熟悉贡献协议后,即可按照 NNI 开发人员教程,创建第一个 PR =) 了:

其它代码库和参考

经作者许可的一些 NNI 用法示例和相关文档。

反馈

加入聊天组:
Gitter 微信
Gitter NNI 微信

相关项目

以探索先进技术和开放为目标,Microsoft Research (MSR) 还发布了一些相关的开源项目。

我们鼓励研究人员和学生利用这些项目来加速 AI 开发和研究。

许可协议

代码库遵循 MIT 许可协议

{% endblock %}