├── LICENSE ├── README.md ├── cal_bit.py ├── cal_bit_huffman.py ├── cifar10_models ├── __init__.py ├── data.py ├── densenet.py ├── googlenet.py ├── inception.py ├── mobilenetv2.py ├── resnet.py ├── resnet_orig.py └── vgg.py ├── clib ├── __init__.py ├── cos_stat │ ├── __init__.py │ ├── cos_stat.cpp │ ├── cos_stat.cu │ ├── cos_stat.py │ └── setup.py ├── lin_stat │ ├── __init__.py │ ├── lin_stat.cpp │ ├── lin_stat.cu │ ├── lin_stat.py │ └── setup.py └── tri_stat │ ├── __init__.py │ ├── setup.py │ ├── tri_stat.cpp │ ├── tri_stat.cu │ └── tri_stat.py ├── common ├── __init__.py └── tools.py ├── compress ├── __init__.py ├── huffman.py └── range_coder.py ├── data_new ├── __init__.py ├── data.py └── imagenet.py ├── model ├── __init__.py ├── mnasnet.py ├── mobilenetv2.py ├── regnet.py ├── resnet.py └── vision_transformer.py ├── quant ├── __init__.py ├── linspace_centric.py └── soft_quant.py ├── requirements.txt ├── run.sh ├── sparse └── soft_sparse.py ├── train.txt ├── train ├── __init__.py ├── logger.py ├── ops.py ├── test_baseline.py ├── train_final.py └── train_yolo.py └── transform ├── edge_scale.py ├── exp.py ├── log.py ├── ms.py └── scale.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/README.md -------------------------------------------------------------------------------- /cal_bit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cal_bit.py -------------------------------------------------------------------------------- /cal_bit_huffman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cal_bit_huffman.py -------------------------------------------------------------------------------- /cifar10_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cifar10_models/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cifar10_models/data.py -------------------------------------------------------------------------------- /cifar10_models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cifar10_models/densenet.py -------------------------------------------------------------------------------- /cifar10_models/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cifar10_models/googlenet.py -------------------------------------------------------------------------------- /cifar10_models/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cifar10_models/inception.py -------------------------------------------------------------------------------- /cifar10_models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cifar10_models/mobilenetv2.py -------------------------------------------------------------------------------- /cifar10_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cifar10_models/resnet.py -------------------------------------------------------------------------------- /cifar10_models/resnet_orig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cifar10_models/resnet_orig.py -------------------------------------------------------------------------------- /cifar10_models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/cifar10_models/vgg.py -------------------------------------------------------------------------------- /clib/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /clib/cos_stat/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /clib/cos_stat/cos_stat.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/cos_stat/cos_stat.cpp -------------------------------------------------------------------------------- /clib/cos_stat/cos_stat.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/cos_stat/cos_stat.cu -------------------------------------------------------------------------------- /clib/cos_stat/cos_stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/cos_stat/cos_stat.py -------------------------------------------------------------------------------- /clib/cos_stat/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/cos_stat/setup.py -------------------------------------------------------------------------------- /clib/lin_stat/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /clib/lin_stat/lin_stat.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/lin_stat/lin_stat.cpp -------------------------------------------------------------------------------- /clib/lin_stat/lin_stat.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/lin_stat/lin_stat.cu -------------------------------------------------------------------------------- /clib/lin_stat/lin_stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/lin_stat/lin_stat.py -------------------------------------------------------------------------------- /clib/lin_stat/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/lin_stat/setup.py -------------------------------------------------------------------------------- /clib/tri_stat/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /clib/tri_stat/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/tri_stat/setup.py -------------------------------------------------------------------------------- /clib/tri_stat/tri_stat.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/tri_stat/tri_stat.cpp -------------------------------------------------------------------------------- /clib/tri_stat/tri_stat.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/tri_stat/tri_stat.cu -------------------------------------------------------------------------------- /clib/tri_stat/tri_stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/clib/tri_stat/tri_stat.py -------------------------------------------------------------------------------- /common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/common/tools.py -------------------------------------------------------------------------------- /compress/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /compress/huffman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/compress/huffman.py -------------------------------------------------------------------------------- /compress/range_coder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/compress/range_coder.py -------------------------------------------------------------------------------- /data_new/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_new/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/data_new/data.py -------------------------------------------------------------------------------- /data_new/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/data_new/imagenet.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/mnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/model/mnasnet.py -------------------------------------------------------------------------------- /model/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/model/mobilenetv2.py -------------------------------------------------------------------------------- /model/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/model/regnet.py -------------------------------------------------------------------------------- /model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/model/resnet.py -------------------------------------------------------------------------------- /model/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/model/vision_transformer.py -------------------------------------------------------------------------------- /quant/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /quant/linspace_centric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/quant/linspace_centric.py -------------------------------------------------------------------------------- /quant/soft_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/quant/soft_quant.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/run.sh -------------------------------------------------------------------------------- /sparse/soft_sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/sparse/soft_sparse.py -------------------------------------------------------------------------------- /train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/train.txt -------------------------------------------------------------------------------- /train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/train/logger.py -------------------------------------------------------------------------------- /train/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/train/ops.py -------------------------------------------------------------------------------- /train/test_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/train/test_baseline.py -------------------------------------------------------------------------------- /train/train_final.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/train/train_final.py -------------------------------------------------------------------------------- /train/train_yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/train/train_yolo.py -------------------------------------------------------------------------------- /transform/edge_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/transform/edge_scale.py -------------------------------------------------------------------------------- /transform/exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/transform/exp.py -------------------------------------------------------------------------------- /transform/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/transform/log.py -------------------------------------------------------------------------------- /transform/ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/transform/ms.py -------------------------------------------------------------------------------- /transform/scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/L2_Compression/HEAD/transform/scale.py --------------------------------------------------------------------------------