The tool manages automated machine learning (AutoML) experiments, dispatches and runs experiments' trial jobs generated by tuning algorithms to search the best neural architecture and/or hyper-parameters in different training environments like Local Machine, Remote Servers, OpenPAI, Kubeflow, FrameworkController on K8S (AKS etc.), DLWorkspace (aka. DLTS), AML (Azure Machine Learning), AdaptDL (aka. ADL), other cloud options and even Hybrid mode.
NNI provides CommandLine Tool as well as an user friendly WebUI to manage training experiements. With the extensible API, you can customize your own AutoML algorithms and training services. To make it easy for new users, NNI also provides a set of build-in stat-of-the-art AutoML algorithms and out of box support for popular training platforms.
Within the following table, we summarized the current NNI capabilities, we are gradually adding new capabilities and we'd love to have your contribution.
| Frameworks & Libraries | Algorithms | Training Services | |
| Built-in |
|
Hyperparameter Tuning
Exhaustive search
Heuristic search
Bayesian optimization
Pruning
Quantization
|
|
| References |
pip install
in an environment that has python 64-bit >= 3.6.
Note:
INFO: Starting restful server...
INFO: Successfully started Restful server!
INFO: Setting local config...
INFO: Successfully set local config!
INFO: Starting experiment...
INFO: Successfully started experiment!
-----------------------------------------------------------------------
The experiment id is egchD4qy
The Web UI urls are: http://223.255.255.1:8080 http://127.0.0.1:8080
-----------------------------------------------------------------------
You can use these commands to get more information about the experiment
-----------------------------------------------------------------------
commands description
1. nnictl experiment show show the information of experiments
2. nnictl trial ls list all of trial jobs
3. nnictl top monitor the status of running experiments
4. nnictl log stderr show stderr log content
5. nnictl log stdout show stdout log content
6. nnictl stop stop an experiment
7. nnictl trial kill kill a trial job by id
8. nnictl --help get help information about nnictl
-----------------------------------------------------------------------
| Gitter | ||
|---|---|---|
|
OR |
|
| Type | Status |
|---|---|
| Fast test |
|
| Full linux |
|
| Full windows |
|
| Type | Status |
|---|---|
| Remote - linux to linux |
|
| Remote - linux to windows |
|
| Remote - windows to linux |
|
| OpenPAI |
|
| Frameworkcontroller |
|
| Kubeflow |
|
| Hybrid |
|
| AzureML |
|
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.
The entire codebase is under MIT license