├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── assets └── pipeline.png ├── configs ├── exp_configs │ ├── rn50_imagenet_baseline.yaml │ └── rn50_imagenet_prune.yaml └── prune_configs │ ├── rn50_fmap.json │ ├── rn50_prune_groups.json │ └── rn50_prune_layer.json ├── data └── dataloaders.py ├── main.py ├── models ├── __init__.py ├── create_model.py ├── resnet.py └── resnet_pruned.py ├── multiproc.py ├── profile.py ├── prune ├── cost.py ├── importance.py ├── prune_config.py └── pruner.py ├── train ├── lr_schedule.py ├── optimizer.py └── training.py └── utils ├── mixup.py ├── model_summary.py ├── smoothing.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | output/ 2 | __pycache__ 3 | *.pkl 4 | *.DS_Store -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/README.md -------------------------------------------------------------------------------- /assets/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/assets/pipeline.png -------------------------------------------------------------------------------- /configs/exp_configs/rn50_imagenet_baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/configs/exp_configs/rn50_imagenet_baseline.yaml -------------------------------------------------------------------------------- /configs/exp_configs/rn50_imagenet_prune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/configs/exp_configs/rn50_imagenet_prune.yaml -------------------------------------------------------------------------------- /configs/prune_configs/rn50_fmap.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/configs/prune_configs/rn50_fmap.json -------------------------------------------------------------------------------- /configs/prune_configs/rn50_prune_groups.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/configs/prune_configs/rn50_prune_groups.json -------------------------------------------------------------------------------- /configs/prune_configs/rn50_prune_layer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/configs/prune_configs/rn50_prune_layer.json -------------------------------------------------------------------------------- /data/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/data/dataloaders.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/create_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/models/create_model.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_pruned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/models/resnet_pruned.py -------------------------------------------------------------------------------- /multiproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/multiproc.py -------------------------------------------------------------------------------- /profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/profile.py -------------------------------------------------------------------------------- /prune/cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/prune/cost.py -------------------------------------------------------------------------------- /prune/importance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/prune/importance.py -------------------------------------------------------------------------------- /prune/prune_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/prune/prune_config.py -------------------------------------------------------------------------------- /prune/pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/prune/pruner.py -------------------------------------------------------------------------------- /train/lr_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/train/lr_schedule.py -------------------------------------------------------------------------------- /train/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/train/optimizer.py -------------------------------------------------------------------------------- /train/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/train/training.py -------------------------------------------------------------------------------- /utils/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/utils/mixup.py -------------------------------------------------------------------------------- /utils/model_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/utils/model_summary.py -------------------------------------------------------------------------------- /utils/smoothing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/utils/smoothing.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/HALP/HEAD/utils/utils.py --------------------------------------------------------------------------------