You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Add export_probs to monitor the architecture weights.
Rewrite configure_optimizers, functions to step optimizers / schedulers, along with other hooks for simplicity, and to be compatible with latest lightning (v1.7).
Align implementation of DifferentiableCell with DARTS official repo.
Re-implementation of ProxylessNAS.
Move nni.retiarii code-base to nni.nas.
Bug fixes
Fix a performance issue caused by tensor formatting in weighted_sum.
Fix a misuse of lambda expression in NAS-Bench-201 search space.
Fix the gumbel temperature schedule in Gumbel DARTS.
Fix the architecture weight sharing when sharing labels in differentiable strategies.
Fix the memo reusing in exporting differentiable cell.
Compression
New tutorial of pruning transformer model. (tutorial)
Add TorchEvaluator, LightningEvaluator, TransformersEvaluator to ease the expression of training logic in pruner. (doc, API)
Enhancements
Promote all pruner API using Evaluator, the old API is deprecated and will be removed in v3.0. (doc)
Greatly enlarge the set of supported operators in pruning speedup via automatic operator conversion.
Support lr_scheduler in pruning by using Evaluator.
Support pruning NLP task in ActivationAPoZRankPruner and ActivationMeanRankPruner.
Add training_steps, regular_scale, movement_mode, sparse_granularity for MovementPruner. (doc)