Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

AutoML for model compression #2573

Merged
merged 125 commits into from
Aug 12, 2020
Merged

AutoML for model compression #2573

merged 125 commits into from
Aug 12, 2020

Conversation

chicm-ms
Copy link
Contributor

@chicm-ms chicm-ms commented Jun 17, 2020

  • NNI pruning framework updates:
    • support channel round (for example, preserved channels up round to 4)
    • support reset wrapper and weights
    • change compressing module ... log message from info to debug
  • AMC pruner implementation
    • agent(DDPG), reuse AMC code
    • pruning environment (use NNI framework to prune)
    • export (speedup)
    • mobilenetv2
  • Example
    • search sparsity and prune model with best sparsity
    • train a pretrained model before search
    • finetune pruned model
  • Benchmark
  • Documentation
  • Unit test

chicm-ms added 30 commits August 6, 2019 11:19
Filter prune algo implementation (microsoft#1655)
document the dispatcher working dir (microsoft#1866)
@ultmaster ultmaster merged commit e9f3cdd into microsoft:master Aug 12, 2020
LovPe pushed a commit to LovPe/nni that referenced this pull request Aug 17, 2020
@chicm-ms chicm-ms deleted the amc branch October 19, 2020 16:00
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants