github microsoft/nni v2.0
NNI v2.0 Release

latest releases: v3.0, v2.10.1, v3.0rc1...
3 years ago

Major updates

Neural architecture search

Training service

  • Support hybrid training service (#3097 #3251 #3252)
  • Support AdlTrainingService, a new training service based on Kubernetes (#3022, thanks external contributors Petuum @pw2393)

Model compression

  • Support pruning schedule for fpgm pruning algorithm (#3110)
  • ModelSpeedup improvement: support torch v1.7 (updated graph_utils.py) (#3076)
  • Improve model compression utility: model flops counter (#3048 #3265)

WebUI & nnictl

  • Support experiments management on WebUI, add a web page for it (#3081 #3127)
  • Improve the layout of overview page (#3046 #3123)
  • Add navigation bar on the right for logs and configs; add expanded icons for table (#3069 #3103)

Others

  • Support launching an experiment from Python code (#3111 #3210 #3263)
  • Refactor builtin/customized tuner installation (#3134)
  • Support new experiment configuration V2 (#3138 #3248 #3251)
  • Reorganize source code directory hierarchy (#2962 #2987 #3037)
  • Change SIGKILL to SIGTERM in local mode when cancelling trial jobs (#3173)
  • Refector hyperband (#3040)

Documentation

  • Port markdown docs to reStructuredText docs and introduce githublink (#3107)
  • List related research and publications in doc (#3150)
  • Add tutorial of saving and loading quantized model (#3192)
  • Remove paiYarn doc and add description of reuse config in remote mode (#3253)
  • Update EfficientNet doc to clarify repo versions (#3158, thanks external contributor @ahundt)

Bug fixes

  • Fix exp-duration pause timing under NO_MORE_TRIAL status (#3043)
  • Fix bug in NAS SPOS trainer, apply_fixed_architecture (#3051, thanks external contributor @HeekangPark)
  • Fix _compute_hessian bug in NAS DARTS (PyTorch version) (#3058, thanks external contributor @hroken)
  • Fix bug of conv1d in the cdarts utils (#3073, thanks external contributor @athaker)
  • Fix the handling of unknown trials when resuming an experiment (#3096)
  • Fix bug of kill command under Windows (#3106)
  • Fix lazy logging (#3108, thanks external contributor @HarshCasper)
  • Fix checkpoint load and save issue in QAT quantizer (#3124, thanks external contributor @eedalong)
  • Fix quant grad function calculation error (#3160, thanks external contributor @eedalong)
  • Fix device assignment bug in quantization algorithm (#3212, thanks external contributor @eedalong)
  • Fix bug in ModelSpeedup and enhance UT for it (#3279)
  • and others

Don't miss a new nni release

NewReleases is sending notifications on new releases.