├── .gitignore ├── .idea └── vcs.xml ├── LICENSE ├── README.md ├── __init__.py ├── base_config.py ├── base_model ├── __init__.py ├── cfqk.py ├── lenet5.py ├── mobilenetv1.py ├── resnet.py ├── stagewise_resnet.py ├── vgg.py └── wrn.py ├── constants.py ├── csgd ├── csgd_pipeline.py ├── csgd_prune.py ├── ddp_csgd_train.py ├── do_csgd.py └── flops_scripts.py ├── custom_layers ├── crop_layer.py ├── flatten_layer.py ├── max_layer.py ├── pad_layer.py ├── scale_layer.py └── se_block.py ├── data ├── data_factory.py ├── dataset_util.py └── imagenet_data.py ├── deprecated ├── __init__.py ├── base_config.py ├── base_model │ ├── __init__.py │ ├── lenet5.py │ ├── mobilenetv1.py │ ├── resnet.py │ └── vgg.py ├── builder.py ├── constants.py ├── csgd │ ├── csgd_pipeline.py │ ├── csgd_prune.py │ ├── csgd_rc56.py │ └── csgd_train.py ├── custom_layers │ ├── flatten_layer.py │ └── se_block.py ├── dataset.py ├── ding_test.py ├── ding_train.py ├── display_hdf5.py ├── model_map.py ├── tensorflow │ ├── cr_base.py │ ├── cr_dc40.py │ ├── csgd_standalone.py │ ├── csgd_utils.py │ ├── mg_train.py │ ├── std_dc40_9382.hdf5 │ ├── tf_dataset.py │ ├── tf_utils.py │ ├── tfm_builder_densenet.py │ ├── tfm_callbacks.py │ ├── tfm_constants.py │ ├── tfm_image_processor.py │ ├── tfm_model.py │ ├── tfm_origin_eval.py │ └── tfm_origin_train.py └── utils │ ├── checkpoint.py │ ├── comm.py │ ├── engine.py │ ├── logger.py │ ├── loss.py │ ├── lr_scheduler.py │ ├── misc.py │ ├── pyt_utils.py │ ├── timer.py │ └── torch_utils.py ├── display_hdf5.py ├── model_map.py ├── ndp_test.py ├── ndp_train.py ├── nobn_builder.py ├── show_log.py ├── train_base_model.py ├── transform_torchvision.py └── utils ├── checkpoint.py ├── comm.py ├── engine.py ├── logger.py ├── loss.py ├── lr_scheduler.py ├── misc.py ├── pyt_utils.py ├── timer.py └── torch_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /base_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/base_config.py -------------------------------------------------------------------------------- /base_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /base_model/cfqk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/base_model/cfqk.py -------------------------------------------------------------------------------- /base_model/lenet5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/base_model/lenet5.py -------------------------------------------------------------------------------- /base_model/mobilenetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/base_model/mobilenetv1.py -------------------------------------------------------------------------------- /base_model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/base_model/resnet.py -------------------------------------------------------------------------------- /base_model/stagewise_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/base_model/stagewise_resnet.py -------------------------------------------------------------------------------- /base_model/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/base_model/vgg.py -------------------------------------------------------------------------------- /base_model/wrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/base_model/wrn.py -------------------------------------------------------------------------------- /constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/constants.py -------------------------------------------------------------------------------- /csgd/csgd_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/csgd/csgd_pipeline.py -------------------------------------------------------------------------------- /csgd/csgd_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/csgd/csgd_prune.py -------------------------------------------------------------------------------- /csgd/ddp_csgd_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/csgd/ddp_csgd_train.py -------------------------------------------------------------------------------- /csgd/do_csgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/csgd/do_csgd.py -------------------------------------------------------------------------------- /csgd/flops_scripts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/csgd/flops_scripts.py -------------------------------------------------------------------------------- /custom_layers/crop_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/custom_layers/crop_layer.py -------------------------------------------------------------------------------- /custom_layers/flatten_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/custom_layers/flatten_layer.py -------------------------------------------------------------------------------- /custom_layers/max_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/custom_layers/max_layer.py -------------------------------------------------------------------------------- /custom_layers/pad_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/custom_layers/pad_layer.py -------------------------------------------------------------------------------- /custom_layers/scale_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/custom_layers/scale_layer.py -------------------------------------------------------------------------------- /custom_layers/se_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/custom_layers/se_block.py -------------------------------------------------------------------------------- /data/data_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/data/data_factory.py -------------------------------------------------------------------------------- /data/dataset_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/data/dataset_util.py -------------------------------------------------------------------------------- /data/imagenet_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/data/imagenet_data.py -------------------------------------------------------------------------------- /deprecated/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deprecated/base_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/base_config.py -------------------------------------------------------------------------------- /deprecated/base_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deprecated/base_model/lenet5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/base_model/lenet5.py -------------------------------------------------------------------------------- /deprecated/base_model/mobilenetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/base_model/mobilenetv1.py -------------------------------------------------------------------------------- /deprecated/base_model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/base_model/resnet.py -------------------------------------------------------------------------------- /deprecated/base_model/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/base_model/vgg.py -------------------------------------------------------------------------------- /deprecated/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/builder.py -------------------------------------------------------------------------------- /deprecated/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/constants.py -------------------------------------------------------------------------------- /deprecated/csgd/csgd_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/csgd/csgd_pipeline.py -------------------------------------------------------------------------------- /deprecated/csgd/csgd_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/csgd/csgd_prune.py -------------------------------------------------------------------------------- /deprecated/csgd/csgd_rc56.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/csgd/csgd_rc56.py -------------------------------------------------------------------------------- /deprecated/csgd/csgd_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/csgd/csgd_train.py -------------------------------------------------------------------------------- /deprecated/custom_layers/flatten_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/custom_layers/flatten_layer.py -------------------------------------------------------------------------------- /deprecated/custom_layers/se_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/custom_layers/se_block.py -------------------------------------------------------------------------------- /deprecated/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/dataset.py -------------------------------------------------------------------------------- /deprecated/ding_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/ding_test.py -------------------------------------------------------------------------------- /deprecated/ding_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/ding_train.py -------------------------------------------------------------------------------- /deprecated/display_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/display_hdf5.py -------------------------------------------------------------------------------- /deprecated/model_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/model_map.py -------------------------------------------------------------------------------- /deprecated/tensorflow/cr_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/cr_base.py -------------------------------------------------------------------------------- /deprecated/tensorflow/cr_dc40.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/cr_dc40.py -------------------------------------------------------------------------------- /deprecated/tensorflow/csgd_standalone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/csgd_standalone.py -------------------------------------------------------------------------------- /deprecated/tensorflow/csgd_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/csgd_utils.py -------------------------------------------------------------------------------- /deprecated/tensorflow/mg_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/mg_train.py -------------------------------------------------------------------------------- /deprecated/tensorflow/std_dc40_9382.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/std_dc40_9382.hdf5 -------------------------------------------------------------------------------- /deprecated/tensorflow/tf_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tf_dataset.py -------------------------------------------------------------------------------- /deprecated/tensorflow/tf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tf_utils.py -------------------------------------------------------------------------------- /deprecated/tensorflow/tfm_builder_densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tfm_builder_densenet.py -------------------------------------------------------------------------------- /deprecated/tensorflow/tfm_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tfm_callbacks.py -------------------------------------------------------------------------------- /deprecated/tensorflow/tfm_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tfm_constants.py -------------------------------------------------------------------------------- /deprecated/tensorflow/tfm_image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tfm_image_processor.py -------------------------------------------------------------------------------- /deprecated/tensorflow/tfm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tfm_model.py -------------------------------------------------------------------------------- /deprecated/tensorflow/tfm_origin_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tfm_origin_eval.py -------------------------------------------------------------------------------- /deprecated/tensorflow/tfm_origin_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/tensorflow/tfm_origin_train.py -------------------------------------------------------------------------------- /deprecated/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/checkpoint.py -------------------------------------------------------------------------------- /deprecated/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/comm.py -------------------------------------------------------------------------------- /deprecated/utils/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/engine.py -------------------------------------------------------------------------------- /deprecated/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/logger.py -------------------------------------------------------------------------------- /deprecated/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/loss.py -------------------------------------------------------------------------------- /deprecated/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/lr_scheduler.py -------------------------------------------------------------------------------- /deprecated/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/misc.py -------------------------------------------------------------------------------- /deprecated/utils/pyt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/pyt_utils.py -------------------------------------------------------------------------------- /deprecated/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/timer.py -------------------------------------------------------------------------------- /deprecated/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/deprecated/utils/torch_utils.py -------------------------------------------------------------------------------- /display_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/display_hdf5.py -------------------------------------------------------------------------------- /model_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/model_map.py -------------------------------------------------------------------------------- /ndp_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/ndp_test.py -------------------------------------------------------------------------------- /ndp_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/ndp_train.py -------------------------------------------------------------------------------- /nobn_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/nobn_builder.py -------------------------------------------------------------------------------- /show_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/show_log.py -------------------------------------------------------------------------------- /train_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/train_base_model.py -------------------------------------------------------------------------------- /transform_torchvision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/transform_torchvision.py -------------------------------------------------------------------------------- /utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/checkpoint.py -------------------------------------------------------------------------------- /utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/comm.py -------------------------------------------------------------------------------- /utils/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/engine.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/lr_scheduler.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/pyt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/pyt_utils.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/timer.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DingXiaoH/Centripetal-SGD/HEAD/utils/torch_utils.py --------------------------------------------------------------------------------