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

Refactor model compression directory structure #2501

Merged
merged 61 commits into from
May 28, 2020

Conversation

chicm-ms
Copy link
Contributor

@chicm-ms chicm-ms commented May 28, 2020

Refactor model compression directory structure as following to better accommodate more automatic pruning algorithms.

compression
  |--tensorflow
  |--torch
       |--pruning (weight masking algorithms)
       |--quantization
       |--controllers? ( TBD,  control the training process, such as automatic compression,  amc)
       |--speedup
       |--utils

Next steps would be:

  1. add basic controllers such as one shot , iterative controllers to prune models using all existing algorithms.
  2. move algorithm configuration part from pruning package into controller, pruning algorithms can prune specified layers with specified sparsity on demand controlled by controllers.
  3. add auto pruners

chicm-ms added 30 commits August 6, 2019 11:19
Filter prune algo implementation (microsoft#1655)
document the dispatcher working dir (microsoft#1866)
@chicm-ms chicm-ms requested a review from QuanluZhang May 28, 2020 06:52
@chicm-ms chicm-ms requested review from ultmaster and suiguoxin May 28, 2020 07:00
@chicm-ms chicm-ms merged commit 5a911b3 into microsoft:master May 28, 2020
@chicm-ms chicm-ms deleted the refactor_compession_dir 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