├── .gitignore ├── README.md ├── classification ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── cifar100_bs16.py │ │ │ ├── cifar10_bs16.py │ │ │ ├── imagenet_bs32.py │ │ │ ├── imagenet_bs32_pil_resize.py │ │ │ ├── imagenet_bs64.py │ │ │ ├── imagenet_bs64_autoaug.py │ │ │ ├── imagenet_bs64_pil_resize.py │ │ │ ├── imagenet_bs64_swin_224.py │ │ │ ├── imagenet_bs64_swin_384.py │ │ │ ├── pipelines │ │ │ │ ├── auto_aug.py │ │ │ │ └── rand_aug.py │ │ │ └── voc_bs16.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── mobilenet_v2_1x.py │ │ │ ├── mobilenet_v3_large_imagenet.py │ │ │ ├── mobilenet_v3_small_cifar.py │ │ │ ├── mobilenet_v3_small_imagenet.py │ │ │ ├── regnet │ │ │ │ ├── regnetx_1.6gf.py │ │ │ │ ├── regnetx_12gf.py │ │ │ │ ├── regnetx_3.2gf.py │ │ │ │ ├── regnetx_4.0gf.py │ │ │ │ ├── regnetx_400mf.py │ │ │ │ ├── regnetx_6.4gf.py │ │ │ │ ├── regnetx_8.0gf.py │ │ │ │ └── regnetx_800mf.py │ │ │ ├── repvgg-A0_in1k.py │ │ │ ├── repvgg-B3_lbs-mixup_in1k.py │ │ │ ├── resnest101.py │ │ │ ├── resnest200.py │ │ │ ├── resnest269.py │ │ │ ├── resnest50.py │ │ │ ├── resnet101.py │ │ │ ├── resnet101_cifar.py │ │ │ ├── resnet152.py │ │ │ ├── resnet152_cifar.py │ │ │ ├── resnet18.py │ │ │ ├── resnet18_cifar.py │ │ │ ├── resnet34.py │ │ │ ├── resnet34_cifar.py │ │ │ ├── resnet50.py │ │ │ ├── resnet50_cifar.py │ │ │ ├── resnet50_cifar_cutmix.py │ │ │ ├── resnet50_cifar_mixup.py │ │ │ ├── resnet50_cutmix.py │ │ │ ├── resnet50_label_smooth.py │ │ │ ├── resnet50_mixup.py │ │ │ ├── resnetv1d101.py │ │ │ ├── resnetv1d152.py │ │ │ ├── resnetv1d50.py │ │ │ ├── resnext101_32x4d.py │ │ │ ├── resnext101_32x8d.py │ │ │ ├── resnext152_32x4d.py │ │ │ ├── resnext50_32x4d.py │ │ │ ├── seresnet101.py │ │ │ ├── seresnet50.py │ │ │ ├── seresnext101_32x4d.py │ │ │ ├── seresnext50_32x4d.py │ │ │ ├── shufflenet_v1_1x.py │ │ │ ├── shufflenet_v2_1x.py │ │ │ ├── swin_transformer │ │ │ │ ├── base_224.py │ │ │ │ ├── base_384.py │ │ │ │ ├── large_224.py │ │ │ │ ├── large_384.py │ │ │ │ ├── small_224.py │ │ │ │ └── tiny_224.py │ │ │ ├── tnt_s_patch16_224.py │ │ │ ├── vgg11.py │ │ │ ├── vgg11bn.py │ │ │ ├── vgg13.py │ │ │ ├── vgg13bn.py │ │ │ ├── vgg16.py │ │ │ ├── vgg16bn.py │ │ │ ├── vgg19.py │ │ │ ├── vgg19bn.py │ │ │ ├── vit_base_patch16_224_finetune.py │ │ │ ├── vit_base_patch16_224_pretrain.py │ │ │ ├── vit_base_patch16_384_finetune.py │ │ │ ├── vit_base_patch32_384_finetune.py │ │ │ ├── vit_large_patch16_224_finetune.py │ │ │ ├── vit_large_patch16_384_finetune.py │ │ │ └── vit_large_patch32_384_finetune.py │ │ └── schedules │ │ │ ├── cifar10_bs128.py │ │ │ ├── imagenet_bs1024_adamw_swin.py │ │ │ ├── imagenet_bs1024_linearlr_bn_nowd.py │ │ │ ├── imagenet_bs2048.py │ │ │ ├── imagenet_bs2048_AdamW.py │ │ │ ├── imagenet_bs2048_coslr.py │ │ │ ├── imagenet_bs256.py │ │ │ ├── imagenet_bs256_140e.py │ │ │ ├── imagenet_bs256_200e_coslr_warmup.py │ │ │ ├── imagenet_bs256_coslr.py │ │ │ ├── imagenet_bs256_epochstep.py │ │ │ └── imagenet_bs4096_AdamW.py │ ├── regnet │ │ ├── regnet_0.4G_origin.py │ │ ├── regnet_0.4G_pruning.py │ │ ├── regnet_0.8G_origin.py │ │ ├── regnet_0.8G_pruning.py │ │ ├── regnet_1.6G_origin.py │ │ ├── regnet_1.6G_pruning.py │ │ ├── regnet_3.2G_finetune.py │ │ ├── regnet_3.2G_origin.py │ │ ├── regnet_3.2G_pruning.py │ │ └── regnet_6.4G_origin.py │ ├── resnet50 │ │ ├── resnet50_finetune.py │ │ └── resnet50_pruning.py │ └── resnext │ │ ├── resnext50_finetune.py │ │ └── resnext50_pruning.py └── tools │ ├── dist_test.sh │ ├── dist_train.sh │ ├── fisher_pruning_hook │ ├── model_eval.py │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ └── train.py ├── detection ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── cityscapes_detection.py │ │ │ ├── cityscapes_instance.py │ │ │ ├── coco_detection.py │ │ │ ├── coco_instance.py │ │ │ ├── coco_instance_semantic.py │ │ │ ├── deepfashion.py │ │ │ ├── lvis_v0.5_instance.py │ │ │ ├── lvis_v1_instance.py │ │ │ ├── voc0712.py │ │ │ └── wider_face.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ │ ├── cascade_rcnn_r50_fpn.py │ │ │ ├── fast_rcnn_r50_fpn.py │ │ │ ├── faster_rcnn_r50_caffe_c4.py │ │ │ ├── faster_rcnn_r50_caffe_dc5.py │ │ │ ├── faster_rcnn_r50_fpn.py │ │ │ ├── mask_rcnn_r50_caffe_c4.py │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ ├── retinanet_r50_fpn.py │ │ │ ├── rpn_r50_caffe_c4.py │ │ │ ├── rpn_r50_fpn.py │ │ │ └── ssd300.py │ │ └── schedules │ │ │ ├── schedule_1x.py │ │ │ ├── schedule_20e.py │ │ │ └── schedule_2x.py │ ├── atss │ │ ├── atss_finetune.py │ │ └── atss_pruning.py │ ├── faster_rcnn │ │ ├── faster_finetune.py │ │ └── faster_pruning.py │ ├── fsaf │ │ ├── fsaf_finetune.py │ │ └── fsaf_pruning.py │ ├── paa │ │ ├── paa_finetune.py │ │ └── paa_pruning.py │ └── retina │ │ ├── retina_finetune.py │ │ └── retina_pruning.py └── tools │ ├── dist_test.sh │ ├── dist_train.sh │ ├── fisher_pruning_hook │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ └── train.py └── fisher_pruning_hook ├── __init__.py ├── fisher_pruning.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .pyc 2 | __pycache__ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/README.md -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/cifar100_bs16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/cifar100_bs16.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/cifar10_bs16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/cifar10_bs16.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/imagenet_bs32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/imagenet_bs32.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/imagenet_bs32_pil_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/imagenet_bs32_pil_resize.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/imagenet_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/imagenet_bs64.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/imagenet_bs64_autoaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/imagenet_bs64_autoaug.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/imagenet_bs64_pil_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/imagenet_bs64_pil_resize.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/imagenet_bs64_swin_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/imagenet_bs64_swin_224.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/imagenet_bs64_swin_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/imagenet_bs64_swin_384.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/pipelines/auto_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/pipelines/auto_aug.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/pipelines/rand_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/pipelines/rand_aug.py -------------------------------------------------------------------------------- /classification/configs/_base_/datasets/voc_bs16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/datasets/voc_bs16.py -------------------------------------------------------------------------------- /classification/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/mobilenet_v2_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/mobilenet_v2_1x.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/mobilenet_v3_large_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/mobilenet_v3_large_imagenet.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/mobilenet_v3_small_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/mobilenet_v3_small_cifar.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/mobilenet_v3_small_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/mobilenet_v3_small_imagenet.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/regnet/regnetx_1.6gf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/regnet/regnetx_1.6gf.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/regnet/regnetx_12gf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/regnet/regnetx_12gf.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/regnet/regnetx_3.2gf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/regnet/regnetx_3.2gf.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/regnet/regnetx_4.0gf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/regnet/regnetx_4.0gf.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/regnet/regnetx_400mf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/regnet/regnetx_400mf.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/regnet/regnetx_6.4gf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/regnet/regnetx_6.4gf.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/regnet/regnetx_8.0gf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/regnet/regnetx_8.0gf.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/regnet/regnetx_800mf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/regnet/regnetx_800mf.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/repvgg-A0_in1k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/repvgg-A0_in1k.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/repvgg-B3_lbs-mixup_in1k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/repvgg-B3_lbs-mixup_in1k.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnest101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnest101.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnest200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnest200.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnest269.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnest269.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnest50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnest50.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet101.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet101_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet101_cifar.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet152.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet152_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet152_cifar.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet18.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet18_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet18_cifar.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet34.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet34_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet34_cifar.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet50.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet50_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet50_cifar.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet50_cifar_cutmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet50_cifar_cutmix.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet50_cifar_mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet50_cifar_mixup.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet50_cutmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet50_cutmix.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet50_label_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet50_label_smooth.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnet50_mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnet50_mixup.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnetv1d101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnetv1d101.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnetv1d152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnetv1d152.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnetv1d50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnetv1d50.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnext101_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnext101_32x4d.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnext101_32x8d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnext101_32x8d.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnext152_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnext152_32x4d.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/resnext50_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/resnext50_32x4d.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/seresnet101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/seresnet101.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/seresnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/seresnet50.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/seresnext101_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/seresnext101_32x4d.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/seresnext50_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/seresnext50_32x4d.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/shufflenet_v1_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/shufflenet_v1_1x.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/shufflenet_v2_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/shufflenet_v2_1x.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/swin_transformer/base_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/swin_transformer/base_224.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/swin_transformer/base_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/swin_transformer/base_384.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/swin_transformer/large_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/swin_transformer/large_224.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/swin_transformer/large_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/swin_transformer/large_384.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/swin_transformer/small_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/swin_transformer/small_224.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/swin_transformer/tiny_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/swin_transformer/tiny_224.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/tnt_s_patch16_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/tnt_s_patch16_224.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vgg11.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vgg11.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vgg11bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vgg11bn.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vgg13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vgg13.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vgg13bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vgg13bn.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vgg16.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vgg16bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vgg16bn.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vgg19.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vgg19.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vgg19bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vgg19bn.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vit_base_patch16_224_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vit_base_patch16_224_finetune.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vit_base_patch16_224_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vit_base_patch16_224_pretrain.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vit_base_patch16_384_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vit_base_patch16_384_finetune.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vit_base_patch32_384_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vit_base_patch32_384_finetune.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vit_large_patch16_224_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vit_large_patch16_224_finetune.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vit_large_patch16_384_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vit_large_patch16_384_finetune.py -------------------------------------------------------------------------------- /classification/configs/_base_/models/vit_large_patch32_384_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/models/vit_large_patch32_384_finetune.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/cifar10_bs128.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/cifar10_bs128.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs1024_adamw_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs1024_adamw_swin.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs1024_linearlr_bn_nowd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs1024_linearlr_bn_nowd.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs2048.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs2048.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs2048_AdamW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs2048_AdamW.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs2048_coslr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs2048_coslr.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs256.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs256_140e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs256_140e.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs256_200e_coslr_warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs256_200e_coslr_warmup.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs256_coslr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs256_coslr.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs256_epochstep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs256_epochstep.py -------------------------------------------------------------------------------- /classification/configs/_base_/schedules/imagenet_bs4096_AdamW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/_base_/schedules/imagenet_bs4096_AdamW.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_0.4G_origin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_0.4G_origin.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_0.4G_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_0.4G_pruning.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_0.8G_origin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_0.8G_origin.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_0.8G_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_0.8G_pruning.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_1.6G_origin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_1.6G_origin.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_1.6G_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_1.6G_pruning.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_3.2G_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_3.2G_finetune.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_3.2G_origin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_3.2G_origin.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_3.2G_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_3.2G_pruning.py -------------------------------------------------------------------------------- /classification/configs/regnet/regnet_6.4G_origin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/regnet/regnet_6.4G_origin.py -------------------------------------------------------------------------------- /classification/configs/resnet50/resnet50_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/resnet50/resnet50_finetune.py -------------------------------------------------------------------------------- /classification/configs/resnet50/resnet50_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/resnet50/resnet50_pruning.py -------------------------------------------------------------------------------- /classification/configs/resnext/resnext50_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/resnext/resnext50_finetune.py -------------------------------------------------------------------------------- /classification/configs/resnext/resnext50_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/configs/resnext/resnext50_pruning.py -------------------------------------------------------------------------------- /classification/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/tools/dist_test.sh -------------------------------------------------------------------------------- /classification/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/tools/dist_train.sh -------------------------------------------------------------------------------- /classification/tools/fisher_pruning_hook: -------------------------------------------------------------------------------- 1 | ../../fisher_pruning_hook -------------------------------------------------------------------------------- /classification/tools/model_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/tools/model_eval.py -------------------------------------------------------------------------------- /classification/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/tools/slurm_test.sh -------------------------------------------------------------------------------- /classification/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/tools/slurm_train.sh -------------------------------------------------------------------------------- /classification/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/tools/test.py -------------------------------------------------------------------------------- /classification/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/classification/tools/train.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/cityscapes_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/cityscapes_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/cityscapes_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/cityscapes_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/coco_instance_semantic.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/deepfashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/deepfashion.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/lvis_v1_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/voc0712.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/datasets/wider_face.py -------------------------------------------------------------------------------- /detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/cascade_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/cascade_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/fast_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/fast_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/faster_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/mask_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/retinanet_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/retinanet_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/rpn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/rpn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/rpn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/rpn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/ssd300.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/models/ssd300.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /detection/configs/atss/atss_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/atss/atss_finetune.py -------------------------------------------------------------------------------- /detection/configs/atss/atss_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/atss/atss_pruning.py -------------------------------------------------------------------------------- /detection/configs/faster_rcnn/faster_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/faster_rcnn/faster_finetune.py -------------------------------------------------------------------------------- /detection/configs/faster_rcnn/faster_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/faster_rcnn/faster_pruning.py -------------------------------------------------------------------------------- /detection/configs/fsaf/fsaf_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/fsaf/fsaf_finetune.py -------------------------------------------------------------------------------- /detection/configs/fsaf/fsaf_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/fsaf/fsaf_pruning.py -------------------------------------------------------------------------------- /detection/configs/paa/paa_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/paa/paa_finetune.py -------------------------------------------------------------------------------- /detection/configs/paa/paa_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/paa/paa_pruning.py -------------------------------------------------------------------------------- /detection/configs/retina/retina_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/retina/retina_finetune.py -------------------------------------------------------------------------------- /detection/configs/retina/retina_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/configs/retina/retina_pruning.py -------------------------------------------------------------------------------- /detection/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/tools/dist_test.sh -------------------------------------------------------------------------------- /detection/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/tools/dist_train.sh -------------------------------------------------------------------------------- /detection/tools/fisher_pruning_hook: -------------------------------------------------------------------------------- 1 | ../../fisher_pruning_hook -------------------------------------------------------------------------------- /detection/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/tools/slurm_test.sh -------------------------------------------------------------------------------- /detection/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/tools/slurm_train.sh -------------------------------------------------------------------------------- /detection/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/tools/test.py -------------------------------------------------------------------------------- /detection/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/detection/tools/train.py -------------------------------------------------------------------------------- /fisher_pruning_hook/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/fisher_pruning_hook/__init__.py -------------------------------------------------------------------------------- /fisher_pruning_hook/fisher_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/fisher_pruning_hook/fisher_pruning.py -------------------------------------------------------------------------------- /fisher_pruning_hook/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ben-Louis/FisherPruning-Pytorch/HEAD/fisher_pruning_hook/utils.py --------------------------------------------------------------------------------