├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── environment.yml ├── experiments ├── __init__.py ├── data.py ├── eval.py ├── scripts │ ├── imagenet-resnet50.sh │ ├── imagenet-vitb16.sh │ ├── slurm-imagenet-resnet50.sh │ ├── slurm-imagenet-vitb16-16x16.sh │ └── slurm-imagenet-vitb16.sh ├── train.py ├── transforms.py └── utils.py └── spartan ├── __init__.py ├── convert.py ├── modules ├── __init__.py ├── attention.py ├── conv.py ├── linear.py └── masked_module.py ├── sparsifier.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/environment.yml -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/data.py -------------------------------------------------------------------------------- /experiments/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/eval.py -------------------------------------------------------------------------------- /experiments/scripts/imagenet-resnet50.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/scripts/imagenet-resnet50.sh -------------------------------------------------------------------------------- /experiments/scripts/imagenet-vitb16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/scripts/imagenet-vitb16.sh -------------------------------------------------------------------------------- /experiments/scripts/slurm-imagenet-resnet50.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/scripts/slurm-imagenet-resnet50.sh -------------------------------------------------------------------------------- /experiments/scripts/slurm-imagenet-vitb16-16x16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/scripts/slurm-imagenet-vitb16-16x16.sh -------------------------------------------------------------------------------- /experiments/scripts/slurm-imagenet-vitb16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/scripts/slurm-imagenet-vitb16.sh -------------------------------------------------------------------------------- /experiments/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/train.py -------------------------------------------------------------------------------- /experiments/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/transforms.py -------------------------------------------------------------------------------- /experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/experiments/utils.py -------------------------------------------------------------------------------- /spartan/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /spartan/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/spartan/convert.py -------------------------------------------------------------------------------- /spartan/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/spartan/modules/__init__.py -------------------------------------------------------------------------------- /spartan/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/spartan/modules/attention.py -------------------------------------------------------------------------------- /spartan/modules/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/spartan/modules/conv.py -------------------------------------------------------------------------------- /spartan/modules/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/spartan/modules/linear.py -------------------------------------------------------------------------------- /spartan/modules/masked_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/spartan/modules/masked_module.py -------------------------------------------------------------------------------- /spartan/sparsifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/spartan/sparsifier.py -------------------------------------------------------------------------------- /spartan/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/spartan/HEAD/spartan/utils.py --------------------------------------------------------------------------------