An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
Retiarii Framework (NNI NAS 2.0) Beta Release with new features:
Repeat
and Cell
(#3481)Note: there are more exciting features of Retiarii planned in the future releases, please refer to Retiarii Roadmap for more information.
Add new NAS algorithm: Blockwise DNAS FBNet (#3532, thanks the external contributor @alibaba-yiwuyao)
ruamel.yaml
(#3702)export_data_url format
(#3665)optimize_mode
on WebUI (#3731)useActiveGpu
in AML v2 config (#3655)experiment_working_directory
in Retiarii config (#3607)Improve NAS 2.0 (Retiarii) Framework (Alpha Release)
ValueChoice
in LayerChoice
(#3508)ValueChoice
(#3508)here <https://github.com/microsoft/nni/issues/3301>
__ for Retiarii Roadmapnni.experiment
(#3490 #3524 #3545)Improve NAS 2.0 (Retiarii) Framework (Improved Experimental)
ValueChoice
(#3349 #3382)preCommand
and enable pythonPath
for remote training service (#3284 #3410)nnicli
to new Python API nni.experiment
(#3334)nni.experiment
), more aligned with nnictl
(#3419)NoneType
error on jupyter notebook (#3337, thanks external contributor @tczhangzhi)githublink
(#3107)reuse
config in remote mode (#3253)_compute_hessian
bug in NAS DARTS (PyTorch version) (#3058, thanks external contributor @hroken)web_channel
in trial_runner
(#2710)$HOME/nni/experiments
to $HOME/nni-experiments
. If you want to view the experiments created by previous NNI releases, you can move the experiments folders from $HOME/nni/experiments
to $HOME/nni-experiments
manually. (#2686) (#2753)_graph_utils
imported inline (#2675)SimulatedAnnealingPruner
(#2736)Provide NAS Open Benchmarks (NasBench101, NasBench201, NDS) with friendly APIs.
Support Classic NAS (i.e., non-weight-sharing mode) on TensorFlow 2.X.
Improve Model Speedup: track more dependencies among layers and automatically resolve mask conflict, support the speedup of pruned resnet.
Added new pruners, including three auto model pruning algorithms: NetAdapt Pruner, SimulatedAnnealing Pruner, AutoCompress Pruner, and ADMM Pruner.
Added model sensitivity analysis tool to help users find the sensitivity of each layer to the pruning.
Update lottery ticket pruner to export winning ticket.
make build
__version__
for SDK version