nni README

The project's README from github.com/microsoft/nni. As one markdown file: README.md.


MIT licensed Issues Bugs Pull Requests Version Documentation Status

NNI automates feature engineering, neural architecture search, hyperparameter tuning, and model compression for deep learning. Find the latest features, API, examples and tutorials in our official documentation (简体中文版点这里).

What's NEW!  

Installation

See the NNI installation guide to install from pip, or build from source.

To install the current release:

$ pip install nni

To update NNI to the latest version, add --upgrade flag to the above commands.

NNI capabilities in a glance

Hyperparameter Tuning Neural Architecture Search Model Compression
Algorithms
Supported Frameworks Training Services Tutorials
Supports
  • PyTorch
  • TensorFlow
  • Scikit-learn
  • XGBoost
  • LightGBM
  • MXNet
  • Caffe2
  • More...
webui

Resources

Contribution guidelines

If you want to contribute to NNI, be sure to review the contribution guidelines, which includes instructions of submitting feedbacks, best coding practices, and code of conduct.

We use GitHub issues to track tracking requests and bugs. Please use NNI Discussion for general questions and new ideas. For questions of specific use cases, please go to Stack Overflow.

Participating discussions via the following IM groups is also welcomed.

Gitter WeChat
image OR image

Over the past few years, NNI has received thousands of feedbacks on GitHub issues, and pull requests from hundreds of contributors. We appreciate all contributions from community to make NNI thrive.

Test status

Essentials

Type Status
Fast test Build Status
Full test - HPO Build Status
Full test - NAS Build Status
Full test - compression Build Status

Training services

Type Status
Local - linux Build Status
Local - windows Build Status
Remote - linux to linux Build Status
Remote - windows to windows Build Status
OpenPAI Build Status
Frameworkcontroller Build Status
Kubeflow Build Status
Hybrid Build Status
AzureML Build Status

Targeting at openness and advancing state-of-art technology, Microsoft Research (MSR) had also released few other open source projects.

We encourage researchers and students leverage these projects to accelerate the AI development and research.

License

The entire codebase is under MIT license.

nni (microsoft/nni)

Searching Database…

14.4K ·GitHub
archived
这是什么?NNI(Neural Network Intelligence)是微软开源的 AutoML 工具包,可以自动化机器学习流程中的多个环节:特征工程、神经网络架构搜索、模型压缩,以及超参数调优。
那它是一个我导入的 Python 库,还是一个更大的平台?它是一个用 pip 安装的 Python 工具包,在你的训练代码里直接调用;但它也能把实验扩展到 Kubernetes、OpenPAI、Kubeflow 或 AzureML 这类训练服务上运行。
它支持哪些机器学习框架?开箱即用就支持 PyTorch、TensorFlow、Scikit-learn、XGBoost、LightGBM 和 MXNet,还有 Caffe2 等——所以你的模型已经在用什么框架,就能直接接上。
它现在还在积极维护吗?不是了——代码仓库已经归档,开发已经停止。最后一次发布是 2022 年 5 月的 v3.0 预览版,别指望有新功能或修复。
Windows 上能用吗,还是只能跑在 Linux 上?都可以:它的测试范围包括 Linux 和 Windows 上的本地训练,也覆盖 Linux 到 Linux、Windows 到 Windows 这类远程场景。
它到底能帮我自动化哪些事?四件事:超参数调优,支持穷举、启发式和贝叶斯搜索;神经网络架构搜索,有多次试验(multi-trial)和一次性(one-shot)两种模式;通过剪枝和量化做模型压缩;再加上特征工程。
它能扩展到我的笔记本之外的集群上吗?可以——支持基于 Kubernetes 的训练服务:OpenPAI、FrameworkController 和 Kubeflow,另外还有 AzureML 和混合部署方案,适合更大规模的调优和 NAS 任务。
有没有视频可以看到实际效果?有——YouTube 上有演示视频,Bilibili 上也有频道(最后更新于 2022 年 6 月);另外还有一场网络研讨会,介绍 NNI 的深度学习探索式训练框架 Retiarii。
Creator
microsoft
Size
130 MB
Stars
14.4K
Age
8y
Commits · 90 Days