├── LICENSE ├── README.md ├── dncnn_stop ├── dataset.py ├── environment.yml ├── final_model_cmd.sh ├── models.py ├── policy_args.py ├── test.py ├── train.py ├── train_stop_joint.py ├── train_stop_kl.py ├── trainer.py └── utils.py ├── lista_stop ├── __init__.py ├── baselines │ ├── __init__.py │ ├── fista.py │ ├── ista.py │ ├── run_fista.sh │ └── run_ista.sh ├── common │ ├── __init__.py │ ├── cmd_args.py │ ├── consts.py │ └── utils.py ├── dataset.py ├── experiments │ ├── __init__.py │ ├── lista.py │ ├── lista_kl.py │ ├── run_lista.sh │ ├── run_lista_stop_stage1.sh │ └── run_lista_stop_stage2.sh ├── model.py └── trainer.py ├── maml_stop ├── .gitignore ├── README.md ├── metal │ ├── __init__.py │ ├── common │ │ ├── __init__.py │ │ ├── cmd_args.py │ │ ├── data_utils.py │ │ ├── meta_modules.py │ │ ├── meta_utils.py │ │ ├── pi_net.py │ │ └── torch_utils.py │ └── experiments │ │ ├── __init__.py │ │ ├── main_opt_stop.py │ │ ├── miniimagenet │ │ ├── __init__.py │ │ ├── run_imba.sh │ │ ├── run_vanilla_5_1.sh │ │ └── run_vanilla_5_5.sh │ │ ├── omniglot │ │ ├── __init__.py │ │ ├── make_split.py │ │ ├── maml_split.txt │ │ ├── run_imba.sh │ │ ├── run_vanilla_20_1.sh │ │ ├── run_vanilla_20_5.sh │ │ ├── run_vanilla_5_1.sh │ │ └── run_vanilla_5_5.sh │ │ ├── opts_callbacks.py │ │ └── utils.py └── setup.py ├── sdn_stop ├── architectures │ ├── CNNs │ │ ├── MobileNet.py │ │ ├── ResNet.py │ │ ├── VGG.py │ │ ├── WideResNet.py │ │ └── __pycache__ │ │ │ ├── MobileNet.cpython-37.pyc │ │ │ ├── ResNet.cpython-37.pyc │ │ │ ├── VGG.cpython-37.pyc │ │ │ └── WideResNet.cpython-37.pyc │ └── SDNs │ │ ├── MobileNet_SDN.py │ │ ├── ResNet_SDN.py │ │ ├── VGG_SDN.py │ │ ├── WideResNet_SDN.py │ │ └── __pycache__ │ │ ├── MobileNet_SDN.cpython-37.pyc │ │ ├── ResNet_SDN.cpython-37.pyc │ │ ├── VGG_SDN.cpython-37.pyc │ │ └── WideResNet_SDN.cpython-37.pyc ├── aux_funcs.py ├── cmd_args.py ├── data.py ├── environment.yml ├── model_funcs.py ├── models.py ├── network_architectures.py ├── profiler.py ├── train_networks.py ├── train_stop_kl.py └── trainer.py └── setup.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/README.md -------------------------------------------------------------------------------- /dncnn_stop/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/dataset.py -------------------------------------------------------------------------------- /dncnn_stop/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/environment.yml -------------------------------------------------------------------------------- /dncnn_stop/final_model_cmd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/final_model_cmd.sh -------------------------------------------------------------------------------- /dncnn_stop/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/models.py -------------------------------------------------------------------------------- /dncnn_stop/policy_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/policy_args.py -------------------------------------------------------------------------------- /dncnn_stop/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/test.py -------------------------------------------------------------------------------- /dncnn_stop/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/train.py -------------------------------------------------------------------------------- /dncnn_stop/train_stop_joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/train_stop_joint.py -------------------------------------------------------------------------------- /dncnn_stop/train_stop_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/train_stop_kl.py -------------------------------------------------------------------------------- /dncnn_stop/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/trainer.py -------------------------------------------------------------------------------- /dncnn_stop/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/dncnn_stop/utils.py -------------------------------------------------------------------------------- /lista_stop/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lista_stop/baselines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lista_stop/baselines/fista.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/baselines/fista.py -------------------------------------------------------------------------------- /lista_stop/baselines/ista.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/baselines/ista.py -------------------------------------------------------------------------------- /lista_stop/baselines/run_fista.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/baselines/run_fista.sh -------------------------------------------------------------------------------- /lista_stop/baselines/run_ista.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/baselines/run_ista.sh -------------------------------------------------------------------------------- /lista_stop/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lista_stop/common/cmd_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/common/cmd_args.py -------------------------------------------------------------------------------- /lista_stop/common/consts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/common/consts.py -------------------------------------------------------------------------------- /lista_stop/common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/common/utils.py -------------------------------------------------------------------------------- /lista_stop/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/dataset.py -------------------------------------------------------------------------------- /lista_stop/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lista_stop/experiments/lista.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/experiments/lista.py -------------------------------------------------------------------------------- /lista_stop/experiments/lista_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/experiments/lista_kl.py -------------------------------------------------------------------------------- /lista_stop/experiments/run_lista.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/experiments/run_lista.sh -------------------------------------------------------------------------------- /lista_stop/experiments/run_lista_stop_stage1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/experiments/run_lista_stop_stage1.sh -------------------------------------------------------------------------------- /lista_stop/experiments/run_lista_stop_stage2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/experiments/run_lista_stop_stage2.sh -------------------------------------------------------------------------------- /lista_stop/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/model.py -------------------------------------------------------------------------------- /lista_stop/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/lista_stop/trainer.py -------------------------------------------------------------------------------- /maml_stop/.gitignore: -------------------------------------------------------------------------------- 1 | *.hdf5 2 | *.egg-info/ 3 | __pycache__/ 4 | scratch/ 5 | *.zip 6 | dataset/ 7 | results/ 8 | -------------------------------------------------------------------------------- /maml_stop/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/README.md -------------------------------------------------------------------------------- /maml_stop/metal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /maml_stop/metal/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /maml_stop/metal/common/cmd_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/common/cmd_args.py -------------------------------------------------------------------------------- /maml_stop/metal/common/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/common/data_utils.py -------------------------------------------------------------------------------- /maml_stop/metal/common/meta_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/common/meta_modules.py -------------------------------------------------------------------------------- /maml_stop/metal/common/meta_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/common/meta_utils.py -------------------------------------------------------------------------------- /maml_stop/metal/common/pi_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/common/pi_net.py -------------------------------------------------------------------------------- /maml_stop/metal/common/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/common/torch_utils.py -------------------------------------------------------------------------------- /maml_stop/metal/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /maml_stop/metal/experiments/main_opt_stop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/main_opt_stop.py -------------------------------------------------------------------------------- /maml_stop/metal/experiments/miniimagenet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /maml_stop/metal/experiments/miniimagenet/run_imba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/miniimagenet/run_imba.sh -------------------------------------------------------------------------------- /maml_stop/metal/experiments/miniimagenet/run_vanilla_5_1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/miniimagenet/run_vanilla_5_1.sh -------------------------------------------------------------------------------- /maml_stop/metal/experiments/miniimagenet/run_vanilla_5_5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/miniimagenet/run_vanilla_5_5.sh -------------------------------------------------------------------------------- /maml_stop/metal/experiments/omniglot/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /maml_stop/metal/experiments/omniglot/make_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/omniglot/make_split.py -------------------------------------------------------------------------------- /maml_stop/metal/experiments/omniglot/maml_split.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/omniglot/maml_split.txt -------------------------------------------------------------------------------- /maml_stop/metal/experiments/omniglot/run_imba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/omniglot/run_imba.sh -------------------------------------------------------------------------------- /maml_stop/metal/experiments/omniglot/run_vanilla_20_1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/omniglot/run_vanilla_20_1.sh -------------------------------------------------------------------------------- /maml_stop/metal/experiments/omniglot/run_vanilla_20_5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/omniglot/run_vanilla_20_5.sh -------------------------------------------------------------------------------- /maml_stop/metal/experiments/omniglot/run_vanilla_5_1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/omniglot/run_vanilla_5_1.sh -------------------------------------------------------------------------------- /maml_stop/metal/experiments/omniglot/run_vanilla_5_5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/omniglot/run_vanilla_5_5.sh -------------------------------------------------------------------------------- /maml_stop/metal/experiments/opts_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/opts_callbacks.py -------------------------------------------------------------------------------- /maml_stop/metal/experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/metal/experiments/utils.py -------------------------------------------------------------------------------- /maml_stop/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/maml_stop/setup.py -------------------------------------------------------------------------------- /sdn_stop/architectures/CNNs/MobileNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/CNNs/MobileNet.py -------------------------------------------------------------------------------- /sdn_stop/architectures/CNNs/ResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/CNNs/ResNet.py -------------------------------------------------------------------------------- /sdn_stop/architectures/CNNs/VGG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/CNNs/VGG.py -------------------------------------------------------------------------------- /sdn_stop/architectures/CNNs/WideResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/CNNs/WideResNet.py -------------------------------------------------------------------------------- /sdn_stop/architectures/CNNs/__pycache__/MobileNet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/CNNs/__pycache__/MobileNet.cpython-37.pyc -------------------------------------------------------------------------------- /sdn_stop/architectures/CNNs/__pycache__/ResNet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/CNNs/__pycache__/ResNet.cpython-37.pyc -------------------------------------------------------------------------------- /sdn_stop/architectures/CNNs/__pycache__/VGG.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/CNNs/__pycache__/VGG.cpython-37.pyc -------------------------------------------------------------------------------- /sdn_stop/architectures/CNNs/__pycache__/WideResNet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/CNNs/__pycache__/WideResNet.cpython-37.pyc -------------------------------------------------------------------------------- /sdn_stop/architectures/SDNs/MobileNet_SDN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/SDNs/MobileNet_SDN.py -------------------------------------------------------------------------------- /sdn_stop/architectures/SDNs/ResNet_SDN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/SDNs/ResNet_SDN.py -------------------------------------------------------------------------------- /sdn_stop/architectures/SDNs/VGG_SDN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/SDNs/VGG_SDN.py -------------------------------------------------------------------------------- /sdn_stop/architectures/SDNs/WideResNet_SDN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/SDNs/WideResNet_SDN.py -------------------------------------------------------------------------------- /sdn_stop/architectures/SDNs/__pycache__/MobileNet_SDN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/SDNs/__pycache__/MobileNet_SDN.cpython-37.pyc -------------------------------------------------------------------------------- /sdn_stop/architectures/SDNs/__pycache__/ResNet_SDN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/SDNs/__pycache__/ResNet_SDN.cpython-37.pyc -------------------------------------------------------------------------------- /sdn_stop/architectures/SDNs/__pycache__/VGG_SDN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/SDNs/__pycache__/VGG_SDN.cpython-37.pyc -------------------------------------------------------------------------------- /sdn_stop/architectures/SDNs/__pycache__/WideResNet_SDN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/architectures/SDNs/__pycache__/WideResNet_SDN.cpython-37.pyc -------------------------------------------------------------------------------- /sdn_stop/aux_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/aux_funcs.py -------------------------------------------------------------------------------- /sdn_stop/cmd_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/cmd_args.py -------------------------------------------------------------------------------- /sdn_stop/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/data.py -------------------------------------------------------------------------------- /sdn_stop/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/environment.yml -------------------------------------------------------------------------------- /sdn_stop/model_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/model_funcs.py -------------------------------------------------------------------------------- /sdn_stop/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/models.py -------------------------------------------------------------------------------- /sdn_stop/network_architectures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/network_architectures.py -------------------------------------------------------------------------------- /sdn_stop/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/profiler.py -------------------------------------------------------------------------------- /sdn_stop/train_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/train_networks.py -------------------------------------------------------------------------------- /sdn_stop/train_stop_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/train_stop_kl.py -------------------------------------------------------------------------------- /sdn_stop/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/sdn_stop/trainer.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xinshi-chen/l2stop/HEAD/setup.py --------------------------------------------------------------------------------