├── .gitignore ├── README.md ├── common ├── __init__.py ├── debug_memory.py ├── io.py ├── time.py ├── timer.py └── utils.py ├── configs ├── cifarnet_retrain.yaml ├── google_resnet50_imagenet_inp-0_fc-80_rest-90_magni.yaml ├── google_resnet50_imagenet_inp-0_fc-80_rest-90_woodfisher.yaml ├── mlpnet_mnist_config_one_shot_woodburry_fisher.yaml ├── mobilenetv1.yaml ├── resnet20_retrain.yaml ├── resnet20_woodfisher.yaml ├── resnet50.yaml └── vgg16.yaml ├── env.py ├── env_cfg.py ├── greedy_alg ├── __init__.py ├── cmpr_objective.py ├── cmpr_pruned.py ├── greedy.py ├── greedy_io.py ├── greedy_options.py ├── greedyblock.py ├── greedyblock.v0.py ├── greedyblock_good.py ├── greedylayer.py ├── mat_utils.py ├── schedule_job.py ├── test.py └── test1.py ├── main.py ├── models ├── VGG.py ├── __init__.py ├── alexnet_pretrain.py ├── cifarnet.py ├── layers │ ├── __init__.py │ ├── mixed_conv.py │ ├── squeeze_exitation_layer.py │ └── utils.py ├── logistic_regression.py ├── mlpnet.py ├── mobilenet.py ├── mobilenet_v1_dropout.py ├── mobilenetv2.py ├── resnet18_cifar.py ├── resnet_cifar10.py ├── resnet_cifar10_swish.py ├── resnet_dpf.py ├── resnet_imagenet.py ├── resnet_mixed_cifar10.py ├── resnet_mixed_imagenet.py ├── resnet_mixed_imagenet_static.py ├── utils.py └── wide_resnet_imagenet.py ├── options.py ├── policies ├── __init__.py ├── manager.py ├── policy.py ├── pruners.py ├── recyclers.py ├── regularizers.py └── trainers.py ├── pruners ├── __init__.py ├── comb_pruner.py ├── diagfisher.py ├── kfac.py ├── magnitude_based.py ├── naivehess.py ├── woodfisher.py ├── woodfisherblock.py ├── woodfisherblock_flops.py ├── woodfisherblockdynamic.py ├── woodfisherblockgroup.py ├── woodtaylor.py └── woodtaylorblock.py ├── scripts ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_multiple.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_single.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_magperb-update_rm_multiple.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_woodfisher-update_rm_multiple.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_woodfisher-update_rm_single.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-mag-update_rm_single.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh ├── cifar10-cifarnet-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh ├── cifar10-cifarnet-backbone_layers-mag.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_selection.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_update_w-scale_update.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_update_w.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_multiple.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_single.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_single_after_each_iteration.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-mag-update_rm_single.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-random-update_rm_multiple.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-random-update_rm_single.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple_with_first_order_term.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh ├── cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_single_with_first_order_term.sh ├── cifar10-vgg16-backbone_layers-mag.sh ├── cifar10_resnet20_tests.comb.sh ├── code-snapshot.sh ├── env.sh ├── find_path.py ├── greedyoffline_update-w_cifar10_resnet20.sh ├── greedyonline_cifar10_resnet20.eval_selection.sh ├── greedyonline_cifar10_resnet20.sh ├── greedyonline_cifar10_resnet20.update_w.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_multiple.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_magperb-update_rm_multiple.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-greedylayer_online_mag_all_layers.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-mag-update_rm_single.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh ├── imagenet-mobilenet-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh ├── imagenet-mobilenet-backbone_layers-mag-gradual_pruning.sh ├── imagenet-mobilenet-backbone_layers-mag.sh ├── imagenet-resnet50-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh ├── imagenet-resnet50-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh ├── mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_mag-update_rm_multiple.sh ├── mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_magperb-update_rm_multiple.sh ├── mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_online_mag_all_layers.sh ├── mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_online_magperb_all_layers.sh ├── mnist-mlpnet-backbone_layers-mag.sh ├── mnist_tests.comb.sh ├── mnist_tests.mag.sh ├── mnist_tests.woodfisher.sh ├── mnist_tests_commands.sh ├── mobilenetv1_global-magni_sparsity-0.7528.sh ├── mobilenetv1_global-magni_sparsity-0.89.sh ├── mobilenetv1_woodfisher_sparsity-0.7528.sh ├── mobilenetv1_woodfisher_sparsity-0.89.sh ├── option.py ├── resnet50_globalmagni_oneshot.sh ├── resnet50_globalmagni_sp-0.8.sh ├── resnet50_globalmagni_sparsity-0.9.sh ├── resnet50_globalmagni_sparsity-0.95.sh ├── resnet50_globalmagni_sparsity-0.98.sh ├── resnet50_woodfisher_sparsity-0.8.sh ├── resnet50_woodfisher_sparsity-0.9.sh ├── resnet50_woodfisher_sparsity-0.95.sh ├── resnet50_woodfisher_sparsity-0.98.sh ├── resnet50_woodfisher_sparsity-0.9_flops-aware-power-0.30.sh ├── resnet50_woodfisher_sparsity-0.9_flops-aware-target-335M.sh ├── resnet50_woodfisher_sparsity-0.9_google-style.sh ├── resnet50_woodfisherblock_oneshot.sh ├── run_all_cifarnet.sh ├── run_batch_baseline.py ├── run_batch_eval.py ├── run_batch_greedy.py ├── run_batch_scale_update.py ├── run_batch_update_weights_with_scale.py ├── sweep_cifar10_cifarnet_woodfisher_train_from_sratch.sh ├── sweep_cifar10_cifarnet_woodfisherblock.sh ├── sweep_cifar10_cifarnet_woodfisherblock_no_weight_update.sh ├── sweep_cifar10_resnet20_oneshot.magnitude.sh ├── sweep_cifar10_resnet20_oneshot.sh ├── sweep_cifar10_resnet20_oneshot_prune_all_weights.sh ├── sweep_cifar10_resnet20_woodfisher.sh ├── sweep_cifar10_resnet20_woodfisher_all_weights.sh ├── sweep_cifar10_resnet20_woodfisher_train_from_sratch.sh ├── sweep_cifar10_resnet20_woodfisherblock.sh ├── sweep_cifar10_resnet20_woodfisherblock_debug_comb_weight_update.sh ├── sweep_cifar10_resnet20_woodfisherblock_no_weight_update.sh ├── sweep_cifar10_vgg16_woodfisherblock.sh ├── sweep_imagenet_mobilenet_woodfisherblock.sh ├── sweep_imagenet_mobilenet_woodfisherblock_no_weight_update.sh ├── sweep_imagenet_resnet50_woodfisherblock.sh ├── sweep_imagenet_resnet50_woodfisherblock_no_weight_update.sh ├── sweep_mlpnet_kfac_aug_12.sh ├── sweep_mlpnet_oneshot.sh ├── sweep_mnist_mlpnet_woodfisher.sh ├── sweep_mnist_mlpnet_woodfisher_no_weight_update.sh └── sweep_resnet50_lobs_comparison_oneshot.sh ├── setup.sh └── utils ├── __init__.py ├── approximation.py ├── auto_augmentation.py ├── checkpoints.py ├── cifar10_log.py ├── csv_writer.py ├── datasets.py ├── flop_utils.py ├── kfac_fisher_utils.py ├── kfac_utils.py ├── manager_utils.py ├── masking_utils.py ├── parse_config.py ├── plot.py ├── pruner_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/README.md -------------------------------------------------------------------------------- /common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/debug_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/common/debug_memory.py -------------------------------------------------------------------------------- /common/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/common/io.py -------------------------------------------------------------------------------- /common/time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/common/time.py -------------------------------------------------------------------------------- /common/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/common/timer.py -------------------------------------------------------------------------------- /common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/common/utils.py -------------------------------------------------------------------------------- /configs/cifarnet_retrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/cifarnet_retrain.yaml -------------------------------------------------------------------------------- /configs/google_resnet50_imagenet_inp-0_fc-80_rest-90_magni.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/google_resnet50_imagenet_inp-0_fc-80_rest-90_magni.yaml -------------------------------------------------------------------------------- /configs/google_resnet50_imagenet_inp-0_fc-80_rest-90_woodfisher.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/google_resnet50_imagenet_inp-0_fc-80_rest-90_woodfisher.yaml -------------------------------------------------------------------------------- /configs/mlpnet_mnist_config_one_shot_woodburry_fisher.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/mlpnet_mnist_config_one_shot_woodburry_fisher.yaml -------------------------------------------------------------------------------- /configs/mobilenetv1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/mobilenetv1.yaml -------------------------------------------------------------------------------- /configs/resnet20_retrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/resnet20_retrain.yaml -------------------------------------------------------------------------------- /configs/resnet20_woodfisher.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/resnet20_woodfisher.yaml -------------------------------------------------------------------------------- /configs/resnet50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/resnet50.yaml -------------------------------------------------------------------------------- /configs/vgg16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/configs/vgg16.yaml -------------------------------------------------------------------------------- /env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/env.py -------------------------------------------------------------------------------- /env_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/env_cfg.py -------------------------------------------------------------------------------- /greedy_alg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /greedy_alg/cmpr_objective.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/cmpr_objective.py -------------------------------------------------------------------------------- /greedy_alg/cmpr_pruned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/cmpr_pruned.py -------------------------------------------------------------------------------- /greedy_alg/greedy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/greedy.py -------------------------------------------------------------------------------- /greedy_alg/greedy_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/greedy_io.py -------------------------------------------------------------------------------- /greedy_alg/greedy_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/greedy_options.py -------------------------------------------------------------------------------- /greedy_alg/greedyblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/greedyblock.py -------------------------------------------------------------------------------- /greedy_alg/greedyblock.v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/greedyblock.v0.py -------------------------------------------------------------------------------- /greedy_alg/greedyblock_good.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/greedyblock_good.py -------------------------------------------------------------------------------- /greedy_alg/greedylayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/greedylayer.py -------------------------------------------------------------------------------- /greedy_alg/mat_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/mat_utils.py -------------------------------------------------------------------------------- /greedy_alg/schedule_job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/schedule_job.py -------------------------------------------------------------------------------- /greedy_alg/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/test.py -------------------------------------------------------------------------------- /greedy_alg/test1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/greedy_alg/test1.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/main.py -------------------------------------------------------------------------------- /models/VGG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/VGG.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/alexnet_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/alexnet_pretrain.py -------------------------------------------------------------------------------- /models/cifarnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/cifarnet.py -------------------------------------------------------------------------------- /models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/layers/__init__.py -------------------------------------------------------------------------------- /models/layers/mixed_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/layers/mixed_conv.py -------------------------------------------------------------------------------- /models/layers/squeeze_exitation_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/layers/squeeze_exitation_layer.py -------------------------------------------------------------------------------- /models/layers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/layers/utils.py -------------------------------------------------------------------------------- /models/logistic_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/logistic_regression.py -------------------------------------------------------------------------------- /models/mlpnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/mlpnet.py -------------------------------------------------------------------------------- /models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/mobilenet.py -------------------------------------------------------------------------------- /models/mobilenet_v1_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/mobilenet_v1_dropout.py -------------------------------------------------------------------------------- /models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/mobilenetv2.py -------------------------------------------------------------------------------- /models/resnet18_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/resnet18_cifar.py -------------------------------------------------------------------------------- /models/resnet_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/resnet_cifar10.py -------------------------------------------------------------------------------- /models/resnet_cifar10_swish.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/resnet_cifar10_swish.py -------------------------------------------------------------------------------- /models/resnet_dpf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/resnet_dpf.py -------------------------------------------------------------------------------- /models/resnet_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/resnet_imagenet.py -------------------------------------------------------------------------------- /models/resnet_mixed_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/resnet_mixed_cifar10.py -------------------------------------------------------------------------------- /models/resnet_mixed_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/resnet_mixed_imagenet.py -------------------------------------------------------------------------------- /models/resnet_mixed_imagenet_static.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/resnet_mixed_imagenet_static.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/utils.py -------------------------------------------------------------------------------- /models/wide_resnet_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/models/wide_resnet_imagenet.py -------------------------------------------------------------------------------- /options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/options.py -------------------------------------------------------------------------------- /policies/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/policies/__init__.py -------------------------------------------------------------------------------- /policies/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/policies/manager.py -------------------------------------------------------------------------------- /policies/policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/policies/policy.py -------------------------------------------------------------------------------- /policies/pruners.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/policies/pruners.py -------------------------------------------------------------------------------- /policies/recyclers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/policies/recyclers.py -------------------------------------------------------------------------------- /policies/regularizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/policies/regularizers.py -------------------------------------------------------------------------------- /policies/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/policies/trainers.py -------------------------------------------------------------------------------- /pruners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/__init__.py -------------------------------------------------------------------------------- /pruners/comb_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/comb_pruner.py -------------------------------------------------------------------------------- /pruners/diagfisher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/diagfisher.py -------------------------------------------------------------------------------- /pruners/kfac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/kfac.py -------------------------------------------------------------------------------- /pruners/magnitude_based.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/magnitude_based.py -------------------------------------------------------------------------------- /pruners/naivehess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/naivehess.py -------------------------------------------------------------------------------- /pruners/woodfisher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/woodfisher.py -------------------------------------------------------------------------------- /pruners/woodfisherblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/woodfisherblock.py -------------------------------------------------------------------------------- /pruners/woodfisherblock_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/woodfisherblock_flops.py -------------------------------------------------------------------------------- /pruners/woodfisherblockdynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/woodfisherblockdynamic.py -------------------------------------------------------------------------------- /pruners/woodfisherblockgroup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/woodfisherblockgroup.py -------------------------------------------------------------------------------- /pruners/woodtaylor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/woodtaylor.py -------------------------------------------------------------------------------- /pruners/woodtaylorblock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/pruners/woodtaylorblock.py -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_magperb-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_magperb-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_woodfisher-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_woodfisher-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_woodfisher-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-greedy_woodfisher-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-mag-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-mag-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/cifar10-cifarnet-backbone_layers-mag.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-cifarnet-backbone_layers-mag.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_selection.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_selection.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_update_w-scale_update.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_update_w-scale_update.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_update_w.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_all_layers-eval_update_w.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_single.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_single_after_each_iteration.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers-upate_rm_single_after_each_iteration.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-mag-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-mag-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-random-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-random-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-random-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-random-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple_with_first_order_term.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple_with_first_order_term.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_single_with_first_order_term.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-resnet20-backbone_layers-blockwise_fisher-woodfisher-update_rm_single_with_first_order_term.sh -------------------------------------------------------------------------------- /scripts/cifar10-vgg16-backbone_layers-mag.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10-vgg16-backbone_layers-mag.sh -------------------------------------------------------------------------------- /scripts/cifar10_resnet20_tests.comb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/cifar10_resnet20_tests.comb.sh -------------------------------------------------------------------------------- /scripts/code-snapshot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/code-snapshot.sh -------------------------------------------------------------------------------- /scripts/env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/env.sh -------------------------------------------------------------------------------- /scripts/find_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/find_path.py -------------------------------------------------------------------------------- /scripts/greedyoffline_update-w_cifar10_resnet20.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/greedyoffline_update-w_cifar10_resnet20.sh -------------------------------------------------------------------------------- /scripts/greedyonline_cifar10_resnet20.eval_selection.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/greedyonline_cifar10_resnet20.eval_selection.sh -------------------------------------------------------------------------------- /scripts/greedyonline_cifar10_resnet20.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/greedyonline_cifar10_resnet20.sh -------------------------------------------------------------------------------- /scripts/greedyonline_cifar10_resnet20.update_w.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/greedyonline_cifar10_resnet20.update_w.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_mag-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_magperb-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_magperb-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_magperb_all_layers.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedy_online_woodfisherblock_all_layers.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedylayer_online_mag_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-greedylayer_online_mag_all_layers.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-mag-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-mag-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-woodfisher-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-blockwise_fisher-woodfisher-update_rm_single.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-mag-gradual_pruning.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-mag-gradual_pruning.sh -------------------------------------------------------------------------------- /scripts/imagenet-mobilenet-backbone_layers-mag.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-mobilenet-backbone_layers-mag.sh -------------------------------------------------------------------------------- /scripts/imagenet-resnet50-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-resnet50-backbone_layers-blockwise_fisher-greedy_online_mag_all_layers.sh -------------------------------------------------------------------------------- /scripts/imagenet-resnet50-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/imagenet-resnet50-backbone_layers-blockwise_fisher-mag-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_mag-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_mag-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_magperb-update_rm_multiple.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_magperb-update_rm_multiple.sh -------------------------------------------------------------------------------- /scripts/mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_online_mag_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_online_mag_all_layers.sh -------------------------------------------------------------------------------- /scripts/mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_online_magperb_all_layers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist-mlpnet-backbone_layers-fullfisher_fisher-greedy_online_magperb_all_layers.sh -------------------------------------------------------------------------------- /scripts/mnist-mlpnet-backbone_layers-mag.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist-mlpnet-backbone_layers-mag.sh -------------------------------------------------------------------------------- /scripts/mnist_tests.comb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist_tests.comb.sh -------------------------------------------------------------------------------- /scripts/mnist_tests.mag.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist_tests.mag.sh -------------------------------------------------------------------------------- /scripts/mnist_tests.woodfisher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist_tests.woodfisher.sh -------------------------------------------------------------------------------- /scripts/mnist_tests_commands.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mnist_tests_commands.sh -------------------------------------------------------------------------------- /scripts/mobilenetv1_global-magni_sparsity-0.7528.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mobilenetv1_global-magni_sparsity-0.7528.sh -------------------------------------------------------------------------------- /scripts/mobilenetv1_global-magni_sparsity-0.89.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mobilenetv1_global-magni_sparsity-0.89.sh -------------------------------------------------------------------------------- /scripts/mobilenetv1_woodfisher_sparsity-0.7528.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mobilenetv1_woodfisher_sparsity-0.7528.sh -------------------------------------------------------------------------------- /scripts/mobilenetv1_woodfisher_sparsity-0.89.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/mobilenetv1_woodfisher_sparsity-0.89.sh -------------------------------------------------------------------------------- /scripts/option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/option.py -------------------------------------------------------------------------------- /scripts/resnet50_globalmagni_oneshot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_globalmagni_oneshot.sh -------------------------------------------------------------------------------- /scripts/resnet50_globalmagni_sp-0.8.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_globalmagni_sp-0.8.sh -------------------------------------------------------------------------------- /scripts/resnet50_globalmagni_sparsity-0.9.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_globalmagni_sparsity-0.9.sh -------------------------------------------------------------------------------- /scripts/resnet50_globalmagni_sparsity-0.95.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_globalmagni_sparsity-0.95.sh -------------------------------------------------------------------------------- /scripts/resnet50_globalmagni_sparsity-0.98.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_globalmagni_sparsity-0.98.sh -------------------------------------------------------------------------------- /scripts/resnet50_woodfisher_sparsity-0.8.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_woodfisher_sparsity-0.8.sh -------------------------------------------------------------------------------- /scripts/resnet50_woodfisher_sparsity-0.9.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_woodfisher_sparsity-0.9.sh -------------------------------------------------------------------------------- /scripts/resnet50_woodfisher_sparsity-0.95.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_woodfisher_sparsity-0.95.sh -------------------------------------------------------------------------------- /scripts/resnet50_woodfisher_sparsity-0.98.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_woodfisher_sparsity-0.98.sh -------------------------------------------------------------------------------- /scripts/resnet50_woodfisher_sparsity-0.9_flops-aware-power-0.30.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_woodfisher_sparsity-0.9_flops-aware-power-0.30.sh -------------------------------------------------------------------------------- /scripts/resnet50_woodfisher_sparsity-0.9_flops-aware-target-335M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_woodfisher_sparsity-0.9_flops-aware-target-335M.sh -------------------------------------------------------------------------------- /scripts/resnet50_woodfisher_sparsity-0.9_google-style.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_woodfisher_sparsity-0.9_google-style.sh -------------------------------------------------------------------------------- /scripts/resnet50_woodfisherblock_oneshot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/resnet50_woodfisherblock_oneshot.sh -------------------------------------------------------------------------------- /scripts/run_all_cifarnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/run_all_cifarnet.sh -------------------------------------------------------------------------------- /scripts/run_batch_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/run_batch_baseline.py -------------------------------------------------------------------------------- /scripts/run_batch_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/run_batch_eval.py -------------------------------------------------------------------------------- /scripts/run_batch_greedy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/run_batch_greedy.py -------------------------------------------------------------------------------- /scripts/run_batch_scale_update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/run_batch_scale_update.py -------------------------------------------------------------------------------- /scripts/run_batch_update_weights_with_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/run_batch_update_weights_with_scale.py -------------------------------------------------------------------------------- /scripts/sweep_cifar10_cifarnet_woodfisher_train_from_sratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_cifarnet_woodfisher_train_from_sratch.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_cifarnet_woodfisherblock.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_cifarnet_woodfisherblock.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_cifarnet_woodfisherblock_no_weight_update.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_cifarnet_woodfisherblock_no_weight_update.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_oneshot.magnitude.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_oneshot.magnitude.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_oneshot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_oneshot.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_oneshot_prune_all_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_oneshot_prune_all_weights.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_woodfisher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_woodfisher.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_woodfisher_all_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_woodfisher_all_weights.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_woodfisher_train_from_sratch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_woodfisher_train_from_sratch.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_woodfisherblock.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_woodfisherblock.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_woodfisherblock_debug_comb_weight_update.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_woodfisherblock_debug_comb_weight_update.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_resnet20_woodfisherblock_no_weight_update.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_resnet20_woodfisherblock_no_weight_update.sh -------------------------------------------------------------------------------- /scripts/sweep_cifar10_vgg16_woodfisherblock.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_cifar10_vgg16_woodfisherblock.sh -------------------------------------------------------------------------------- /scripts/sweep_imagenet_mobilenet_woodfisherblock.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_imagenet_mobilenet_woodfisherblock.sh -------------------------------------------------------------------------------- /scripts/sweep_imagenet_mobilenet_woodfisherblock_no_weight_update.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_imagenet_mobilenet_woodfisherblock_no_weight_update.sh -------------------------------------------------------------------------------- /scripts/sweep_imagenet_resnet50_woodfisherblock.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_imagenet_resnet50_woodfisherblock.sh -------------------------------------------------------------------------------- /scripts/sweep_imagenet_resnet50_woodfisherblock_no_weight_update.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_imagenet_resnet50_woodfisherblock_no_weight_update.sh -------------------------------------------------------------------------------- /scripts/sweep_mlpnet_kfac_aug_12.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_mlpnet_kfac_aug_12.sh -------------------------------------------------------------------------------- /scripts/sweep_mlpnet_oneshot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_mlpnet_oneshot.sh -------------------------------------------------------------------------------- /scripts/sweep_mnist_mlpnet_woodfisher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_mnist_mlpnet_woodfisher.sh -------------------------------------------------------------------------------- /scripts/sweep_mnist_mlpnet_woodfisher_no_weight_update.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_mnist_mlpnet_woodfisher_no_weight_update.sh -------------------------------------------------------------------------------- /scripts/sweep_resnet50_lobs_comparison_oneshot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/scripts/sweep_resnet50_lobs_comparison_oneshot.sh -------------------------------------------------------------------------------- /setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/setup.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/approximation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/approximation.py -------------------------------------------------------------------------------- /utils/auto_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/auto_augmentation.py -------------------------------------------------------------------------------- /utils/checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/checkpoints.py -------------------------------------------------------------------------------- /utils/cifar10_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/cifar10_log.py -------------------------------------------------------------------------------- /utils/csv_writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/csv_writer.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/flop_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/flop_utils.py -------------------------------------------------------------------------------- /utils/kfac_fisher_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/kfac_fisher_utils.py -------------------------------------------------------------------------------- /utils/kfac_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/kfac_utils.py -------------------------------------------------------------------------------- /utils/manager_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/manager_utils.py -------------------------------------------------------------------------------- /utils/masking_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/masking_utils.py -------------------------------------------------------------------------------- /utils/parse_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/parse_config.py -------------------------------------------------------------------------------- /utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/plot.py -------------------------------------------------------------------------------- /utils/pruner_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/pruner_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuxwind/CBS/HEAD/utils/utils.py --------------------------------------------------------------------------------