-
- Downloads
AMC: add direct integration of HAN Lab's DDPG agent
Integrate the code for the DDPG agent from: https://github.com/mit-han-lab/amc-release The instructions for cloning HAN's code and then making changes to fit Distiller were too complicated, so we added the integrated files to distiller/examples/auto_compression/amc/rl_lib/hanlab
Showing
- examples/auto_compression/amc/README.md 29 additions, 0 deletionsexamples/auto_compression/amc/README.md
- examples/auto_compression/amc/amc.py 4 additions, 4 deletionsexamples/auto_compression/amc/amc.py
- examples/auto_compression/amc/amc_args.py 1 addition, 1 deletionexamples/auto_compression/amc/amc_args.py
- examples/auto_compression/amc/auto_compression_channels.yaml 1 addition, 1 deletionexamples/auto_compression/amc/auto_compression_channels.yaml
- examples/auto_compression/amc/auto_compression_filters.yaml 1 addition, 1 deletionexamples/auto_compression/amc/auto_compression_filters.yaml
- examples/auto_compression/amc/jupyter/amc_plain20.ipynb 4 additions, 4 deletionsexamples/auto_compression/amc/jupyter/amc_plain20.ipynb
- examples/auto_compression/amc/jupyter/amc_resnet20.ipynb 4 additions, 4 deletionsexamples/auto_compression/amc/jupyter/amc_resnet20.ipynb
- examples/auto_compression/amc/rl_libs/hanlab/README.md 15 additions, 0 deletionsexamples/auto_compression/amc/rl_libs/hanlab/README.md
- examples/auto_compression/amc/rl_libs/hanlab/__init__.py 0 additions, 0 deletionsexamples/auto_compression/amc/rl_libs/hanlab/__init__.py
- examples/auto_compression/amc/rl_libs/hanlab/agent.py 298 additions, 0 deletionsexamples/auto_compression/amc/rl_libs/hanlab/agent.py
- examples/auto_compression/amc/rl_libs/hanlab/hanlab_if.py 3 additions, 3 deletionsexamples/auto_compression/amc/rl_libs/hanlab/hanlab_if.py
- examples/auto_compression/amc/rl_libs/hanlab/memory.py 228 additions, 0 deletionsexamples/auto_compression/amc/rl_libs/hanlab/memory.py
- examples/auto_compression/amc/rl_libs/hanlab/utils.py 261 additions, 0 deletionsexamples/auto_compression/amc/rl_libs/hanlab/utils.py
- licenses/hanlab-amc-license.txt 21 additions, 0 deletionslicenses/hanlab-amc-license.txt
Loading
Please register or sign in to comment