├── README.md ├── VGG_cifar ├── PFEC_finetune.py ├── PFEC_vggprune.py ├── __init__.py ├── main_cifar_vgg.py ├── main_cifar_vgg_log.py ├── pruning_cifar_vgg.py └── scripts │ ├── PFEC_train_prune.sh │ └── pruning_vgg_my_method.sh ├── functions ├── cal_flop.py ├── explain.png ├── infer_pruned.py └── panda.png ├── models ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── caffe_cifar.cpython-36.pyc │ ├── densenet.cpython-36.pyc │ ├── imagenet_resnet.cpython-36.pyc │ ├── preresnet.cpython-36.pyc │ ├── res_utils.cpython-36.pyc │ ├── resnet.cpython-36.pyc │ ├── resnet_mod.cpython-36.pyc │ ├── resnet_small.cpython-36.pyc │ └── resnext.cpython-36.pyc ├── caffe_cifar.py ├── densenet.py ├── imagenet_resnet.py ├── imagenet_resnet_small.py ├── preresnet.py ├── res_utils.py ├── resnet.py ├── resnet_feature.py ├── resnet_mod.py ├── resnet_small.py ├── resnet_small_V3.py ├── resnext.py ├── vgg.py ├── vgg_cifar.py └── vgg_cifar10.py ├── original_train.py ├── pdf └── FPGM_updated.pdf ├── pruning_cifar10.py ├── pruning_imagenet.py ├── scripts ├── ablation_pruning_cifar10.sh ├── gpu.sh ├── gpu2.sh ├── pruning_cifar10.sh ├── pruning_cifar10_new.sh ├── pruning_imagenet_resnet.sh └── pruning_imagenet_vgg.sh └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/README.md -------------------------------------------------------------------------------- /VGG_cifar/PFEC_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/VGG_cifar/PFEC_finetune.py -------------------------------------------------------------------------------- /VGG_cifar/PFEC_vggprune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/VGG_cifar/PFEC_vggprune.py -------------------------------------------------------------------------------- /VGG_cifar/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/VGG_cifar/__init__.py -------------------------------------------------------------------------------- /VGG_cifar/main_cifar_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/VGG_cifar/main_cifar_vgg.py -------------------------------------------------------------------------------- /VGG_cifar/main_cifar_vgg_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/VGG_cifar/main_cifar_vgg_log.py -------------------------------------------------------------------------------- /VGG_cifar/pruning_cifar_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/VGG_cifar/pruning_cifar_vgg.py -------------------------------------------------------------------------------- /VGG_cifar/scripts/PFEC_train_prune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/VGG_cifar/scripts/PFEC_train_prune.sh -------------------------------------------------------------------------------- /VGG_cifar/scripts/pruning_vgg_my_method.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/VGG_cifar/scripts/pruning_vgg_my_method.sh -------------------------------------------------------------------------------- /functions/cal_flop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/functions/cal_flop.py -------------------------------------------------------------------------------- /functions/explain.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/functions/explain.png -------------------------------------------------------------------------------- /functions/infer_pruned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/functions/infer_pruned.py -------------------------------------------------------------------------------- /functions/panda.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/functions/panda.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/caffe_cifar.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/caffe_cifar.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/densenet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/densenet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/imagenet_resnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/imagenet_resnet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/preresnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/preresnet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/res_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/res_utils.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/resnet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet_mod.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/resnet_mod.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet_small.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/resnet_small.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnext.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/__pycache__/resnext.cpython-36.pyc -------------------------------------------------------------------------------- /models/caffe_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/caffe_cifar.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/imagenet_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/imagenet_resnet.py -------------------------------------------------------------------------------- /models/imagenet_resnet_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/imagenet_resnet_small.py -------------------------------------------------------------------------------- /models/preresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/preresnet.py -------------------------------------------------------------------------------- /models/res_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/res_utils.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/resnet_feature.py -------------------------------------------------------------------------------- /models/resnet_mod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/resnet_mod.py -------------------------------------------------------------------------------- /models/resnet_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/resnet_small.py -------------------------------------------------------------------------------- /models/resnet_small_V3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/resnet_small_V3.py -------------------------------------------------------------------------------- /models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/resnext.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/vgg.py -------------------------------------------------------------------------------- /models/vgg_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/vgg_cifar.py -------------------------------------------------------------------------------- /models/vgg_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/models/vgg_cifar10.py -------------------------------------------------------------------------------- /original_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/original_train.py -------------------------------------------------------------------------------- /pdf/FPGM_updated.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/pdf/FPGM_updated.pdf -------------------------------------------------------------------------------- /pruning_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/pruning_cifar10.py -------------------------------------------------------------------------------- /pruning_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/pruning_imagenet.py -------------------------------------------------------------------------------- /scripts/ablation_pruning_cifar10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/scripts/ablation_pruning_cifar10.sh -------------------------------------------------------------------------------- /scripts/gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/scripts/gpu.sh -------------------------------------------------------------------------------- /scripts/gpu2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/scripts/gpu2.sh -------------------------------------------------------------------------------- /scripts/pruning_cifar10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/scripts/pruning_cifar10.sh -------------------------------------------------------------------------------- /scripts/pruning_cifar10_new.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/scripts/pruning_cifar10_new.sh -------------------------------------------------------------------------------- /scripts/pruning_imagenet_resnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/scripts/pruning_imagenet_resnet.sh -------------------------------------------------------------------------------- /scripts/pruning_imagenet_vgg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/scripts/pruning_imagenet_vgg.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/he-y/filter-pruning-geometric-median/HEAD/utils.py --------------------------------------------------------------------------------