nni
nni copied to clipboard
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
### Description ### Example: - End2End pruning example for `bert` on MNLI dataset and report a few experiments result. Enhancements: - MovementPruner: (Experimental) Support `soft` movement pruning (only support `hard`...
I try to use nni in the HPC at our school. The code is work on my computer. The HPC has many compute nodes and we should submit the tasks...
### Description ### Add fixed subnet state dict load operation and Autoformer evolution ### How to test ### ``` import nni.retiarii.hub.pytorch as hub from nni.retiarii.hub.pytorch.utils.fixed import FixedFactory from nni.retiarii.utils import...
### Description ### Add a new tutorial of searching in DARTS space with DARTS strategy. #### Test Options #### - [ ] fast test - [ ] full test -...
- Release manager: @ultmaster - Release start date: 6.22 - Feature freeze date (at most 30 days): ~7.22~ 8.1 - First test package (1 week since feature freeze): ~7.29~ ~8.12~...
### Description ### Add examples for reproducing: * DARTS * NAS-Bench-101 * ProxylessNAS #### Test Options #### - [ ] fast test - [ ] full test - HPO -...
### Description ### #### Features * Add export_probs to monitor the architecture weights. * Rewrite configure_optimizers, functions to step optimzers / schedulers, along with other hooks for simplicity, and to...
### Description ### * Support drop path in NASNet. * Fix auxiliary loss implementation in NASNet. * Fix typecheck issues in ShuffleNet. #### Test Options #### - [ ] fast...
**Describe the issue**: May like issue #4853 **torch.cuda.is_available()** is FALSE at the second trial but TRUE at the first trial (by print). Also, it is TRUE when running the main.py....