Official pytorch code for "APP: Anytime Progressive Pruning"

Last update: Nov 22, 2022

Overview

APP: Anytime Progressive Pruning

Diganta Misra^1,2,3, Bharat Runwal^2,4, Tianlong Chen⁵, Zhangyang Wang⁵, Irina Rish^1,3

¹ Mila - Quebec AI Institute,² Landskape AI,³ UdeM,⁴ IIT-Delhi,⁵ VITA, UT-Austin

Requirements

To create a new conda environment with the dependencies used in this project, do:

conda env create -f app.yml

For running the code on Restricted-Imagenet Dataset, first install the robustness library from here and provide the imagenet_path argument as the path to the imaganet data folder.

Run the Code

Here is an example of running the Anytime Progressive Pruning (APP) on Cifar-10 dataset with 8 megabatches in total:

python main_anytime_train.py \
    --data ../data \
    --dataset cifar10 \
    --arch resnet50 \
    --seed 1 \
    --epochs 50 \
    --decreasing_lr 20,40 \
    --batch_size 64 \
    --weight_decay 1e-4 \
    --meta_batch_size 6250 \
    --meta_batch_number 8 \
    --sparsity_level 4.5 \
    --snip_size 0.20 \
    --save_dir c10_r50

One-Shot pruning :

python main_anytime_one.py \
    --data ../data \
    --dataset cifar10 \
    --arch resnet50 \
    --seed 1 \
    --epochs 50 \
    --decreasing_lr 20,40 \
    --batch_size 64 \
    --weight_decay 1e-4 \
    --meta_batch_size 6250 \
    --meta_batch_number 8 \
    --sparsity_level 4.5 \
    --snip_size 0.20 \
    --save_dir c10_OSP_r18

Baseline :

python main_anytime_baseline.py \
    --data ../data \
    --dataset cifar10 \
    --arch resnet50 \
    --seed 1 \
    --epochs 50 \
    --decreasing_lr 20,40 \
    --batch_size 64 \
    --weight_decay 1e-4 \
    --meta_batch_size 6250 \
    --meta_batch_number 8 \
    --save_dir c10_BASE_r50

Cite:

@misc{misra2022app,
    title={APP: Anytime Progressive Pruning},
    author={Diganta Misra and Bharat Runwal and Tianlong Chen and Zhangyang Wang and Irina Rish},
    year={2022},
    eprint={2204.01640},
    archivePrefix={arXiv},
    primaryClass={cs.LG}
}

Official pytorch code for "APP: Anytime Progressive Pruning"

Related tags

Overview

APP: Anytime Progressive Pruning

Requirements

Run the Code

Cite:

Owner

Landskape AI

Official repository for ABC-GAN

Siamese TabNet

The tl;dr on a few notable transformer/language model papers + other papers (alignment, memorization, etc).

Informal Persian Universal Dependency Treebank

This tutorial aims to learn the basics of deep learning by hands, and master the basics through combination of lectures and exercises

Transformer Tracking (CVPR2021)

A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.

Trading Gym is an open source project for the development of reinforcement learning algorithms in the context of trading.

Machine Learning Toolkit for Kubernetes

Learning To Have An Ear For Face Super-Resolution

Compares various time-series feature sets on computational performance, within-set structure, and between-set relationships.

This repository contains an implementation of ConvMixer for the ICLR 2022 submission "Patches Are All You Need?".

Project of 'TBEFN: A Two-branch Exposure-fusion Network for Low-light Image Enhancement '

Official git for "CTAB-GAN: Effective Table Data Synthesizing"

8-week curriculum for AI Builders

Code release for General Greedy De-bias Learning

AI4Good project for detecting waste in the environment

Unofficial pytorch implementation of the paper "Dynamic High-Pass Filtering and Multi-Spectral Attention for Image Super-Resolution"

ML-Decoder: Scalable and Versatile Classification Head

Benchmarks for the Optimal Power Flow Problem