├── .gitignore ├── LICENSE ├── README.md ├── configs ├── ps_vit_b_14.yaml ├── ps_vit_b_16.yaml ├── ps_vit_b_18.yaml └── ps_vit_ti_14.yaml ├── datasets ├── __init__.py └── list_dataset.py ├── imgs └── overview.png ├── layers ├── __init__.py ├── cinclude │ ├── cpp_helper.hpp │ ├── cuda_helper.hpp │ └── progressive_sampling_cuda_kernel.cuh ├── csrc │ ├── info.cpp │ ├── progressive_sampling.cpp │ ├── progressive_sampling_cuda.cu │ └── pybind.cpp └── progressive_sample.py ├── main.py ├── models ├── __init__.py ├── ps_vit.py └── transformer_block.py ├── scripts ├── train_distributed.sh └── train_slurm.sh ├── setup.py └── utils ├── __init__.py ├── distributed_utils.py ├── ext_loader.py ├── flop_count ├── __init__.py ├── flop_count.py ├── jit_analysis.py └── jit_handles.py ├── loader.py └── sampler.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/README.md -------------------------------------------------------------------------------- /configs/ps_vit_b_14.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/configs/ps_vit_b_14.yaml -------------------------------------------------------------------------------- /configs/ps_vit_b_16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/configs/ps_vit_b_16.yaml -------------------------------------------------------------------------------- /configs/ps_vit_b_18.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/configs/ps_vit_b_18.yaml -------------------------------------------------------------------------------- /configs/ps_vit_ti_14.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/configs/ps_vit_ti_14.yaml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/list_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/datasets/list_dataset.py -------------------------------------------------------------------------------- /imgs/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/imgs/overview.png -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/cinclude/cpp_helper.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/cinclude/cpp_helper.hpp -------------------------------------------------------------------------------- /layers/cinclude/cuda_helper.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/cinclude/cuda_helper.hpp -------------------------------------------------------------------------------- /layers/cinclude/progressive_sampling_cuda_kernel.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/cinclude/progressive_sampling_cuda_kernel.cuh -------------------------------------------------------------------------------- /layers/csrc/info.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/csrc/info.cpp -------------------------------------------------------------------------------- /layers/csrc/progressive_sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/csrc/progressive_sampling.cpp -------------------------------------------------------------------------------- /layers/csrc/progressive_sampling_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/csrc/progressive_sampling_cuda.cu -------------------------------------------------------------------------------- /layers/csrc/pybind.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/csrc/pybind.cpp -------------------------------------------------------------------------------- /layers/progressive_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/layers/progressive_sample.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .ps_vit import * 2 | -------------------------------------------------------------------------------- /models/ps_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/models/ps_vit.py -------------------------------------------------------------------------------- /models/transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/models/transformer_block.py -------------------------------------------------------------------------------- /scripts/train_distributed.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/scripts/train_distributed.sh -------------------------------------------------------------------------------- /scripts/train_slurm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/scripts/train_slurm.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/setup.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/utils/distributed_utils.py -------------------------------------------------------------------------------- /utils/ext_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/utils/ext_loader.py -------------------------------------------------------------------------------- /utils/flop_count/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/flop_count/flop_count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/utils/flop_count/flop_count.py -------------------------------------------------------------------------------- /utils/flop_count/jit_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/utils/flop_count/jit_analysis.py -------------------------------------------------------------------------------- /utils/flop_count/jit_handles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/utils/flop_count/jit_handles.py -------------------------------------------------------------------------------- /utils/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/utils/loader.py -------------------------------------------------------------------------------- /utils/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuexy/PS-ViT/HEAD/utils/sampler.py --------------------------------------------------------------------------------