├── .gitignore ├── datasets ├── .keep └── MNIST │ ├── processed │ ├── test.pt │ └── training.pt │ └── raw │ ├── t10k-images-idx3-ubyte │ ├── t10k-labels-idx1-ubyte │ ├── train-images-idx3-ubyte │ ├── train-images-idx3-ubyte.gz │ └── train-labels-idx1-ubyte ├── kd ├── detector.py ├── models │ ├── student_net.pth │ └── teacher_net.pth ├── nets.py └── trainer.py ├── pruning ├── detector.py ├── models │ ├── net.pth │ ├── pruned_net_with_conv.pth │ ├── pruned_net_with_torch_0.5_l1.pth │ └── pruned_net_without_conv.pth ├── nets.py ├── pruning_net.py ├── test.py ├── trainer.py └── utils.py └── quantization ├── detector.py ├── models ├── net.pth ├── net_convert.pth ├── net_convert_qat.pth └── net_fuse.pth ├── nets.py ├── qat.py ├── quantize.py ├── test.py ├── test1.py └── trainer.py /.gitignore: -------------------------------------------------------------------------------- 1 | # Project exclude paths 2 | /venv/ -------------------------------------------------------------------------------- /datasets/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/MNIST/processed/test.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/datasets/MNIST/processed/test.pt -------------------------------------------------------------------------------- /datasets/MNIST/processed/training.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/datasets/MNIST/processed/training.pt -------------------------------------------------------------------------------- /datasets/MNIST/raw/t10k-images-idx3-ubyte: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/datasets/MNIST/raw/t10k-images-idx3-ubyte -------------------------------------------------------------------------------- /datasets/MNIST/raw/t10k-labels-idx1-ubyte: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/datasets/MNIST/raw/t10k-labels-idx1-ubyte -------------------------------------------------------------------------------- /datasets/MNIST/raw/train-images-idx3-ubyte: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/datasets/MNIST/raw/train-images-idx3-ubyte -------------------------------------------------------------------------------- /datasets/MNIST/raw/train-images-idx3-ubyte.gz: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/MNIST/raw/train-labels-idx1-ubyte: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/datasets/MNIST/raw/train-labels-idx1-ubyte -------------------------------------------------------------------------------- /kd/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/kd/detector.py -------------------------------------------------------------------------------- /kd/models/student_net.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/kd/models/student_net.pth -------------------------------------------------------------------------------- /kd/models/teacher_net.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/kd/models/teacher_net.pth -------------------------------------------------------------------------------- /kd/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/kd/nets.py -------------------------------------------------------------------------------- /kd/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/kd/trainer.py -------------------------------------------------------------------------------- /pruning/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/detector.py -------------------------------------------------------------------------------- /pruning/models/net.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/models/net.pth -------------------------------------------------------------------------------- /pruning/models/pruned_net_with_conv.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/models/pruned_net_with_conv.pth -------------------------------------------------------------------------------- /pruning/models/pruned_net_with_torch_0.5_l1.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/models/pruned_net_with_torch_0.5_l1.pth -------------------------------------------------------------------------------- /pruning/models/pruned_net_without_conv.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/models/pruned_net_without_conv.pth -------------------------------------------------------------------------------- /pruning/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/nets.py -------------------------------------------------------------------------------- /pruning/pruning_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/pruning_net.py -------------------------------------------------------------------------------- /pruning/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/test.py -------------------------------------------------------------------------------- /pruning/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/trainer.py -------------------------------------------------------------------------------- /pruning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/pruning/utils.py -------------------------------------------------------------------------------- /quantization/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/detector.py -------------------------------------------------------------------------------- /quantization/models/net.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/models/net.pth -------------------------------------------------------------------------------- /quantization/models/net_convert.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/models/net_convert.pth -------------------------------------------------------------------------------- /quantization/models/net_convert_qat.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/models/net_convert_qat.pth -------------------------------------------------------------------------------- /quantization/models/net_fuse.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/models/net_fuse.pth -------------------------------------------------------------------------------- /quantization/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/nets.py -------------------------------------------------------------------------------- /quantization/qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/qat.py -------------------------------------------------------------------------------- /quantization/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/quantize.py -------------------------------------------------------------------------------- /quantization/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/test.py -------------------------------------------------------------------------------- /quantization/test1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/test1.py -------------------------------------------------------------------------------- /quantization/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BastianChen/Model-Compression-Demo/HEAD/quantization/trainer.py --------------------------------------------------------------------------------