├── README.md ├── __pycache__ ├── SNNWrapper.cpython-39.pyc ├── engine_finetune.cpython-310.pyc ├── engine_finetune.cpython-39.pyc ├── engine_finetune_snn.cpython-39.pyc ├── models_vit.cpython-310.pyc ├── models_vit.cpython-39.pyc ├── spike_quan_layer.cpython-310.pyc ├── spike_quan_layer.cpython-39.pyc ├── spike_quan_layer_snn.cpython-39.pyc ├── spike_quan_wrapper.cpython-310.pyc ├── spike_quan_wrapper.cpython-39.pyc └── spike_quan_wrapper_ICML.cpython-39.pyc ├── energy_consumption_calculation ├── README.md ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── __init__.cpython-39.pyc │ ├── engine.cpython-310.pyc │ ├── engine.cpython-39.pyc │ ├── flops_counter.cpython-310.pyc │ ├── flops_counter.cpython-39.pyc │ ├── ops.cpython-310.pyc │ ├── ops.cpython-39.pyc │ ├── utils.cpython-310.pyc │ └── utils.cpython-39.pyc ├── engine.py ├── flops_counter.py ├── ops.py ├── test_spikingformer-8-512_eg.log └── utils.py ├── engine_finetune.py ├── engine_pretrain.py ├── extract_imagenet.py ├── main.py ├── main_finetune.py ├── main_finetune_distill.py ├── main_finetune_distill_dvs.py ├── main_finetune_dvs.py ├── main_finetune_raw.py ├── main_linprobe.py ├── main_pretrain.py ├── metric.py ├── misc.py ├── models_mae.py ├── models_vit.py ├── results ├── vit-large-imagenet-q32-83.86.txt └── vit-small-imagenet-q32-81.59.txt ├── scripts ├── vit-base_QAT_distill_32.sh ├── vit-large_QAT_distill_32.sh ├── vit-small_QAT_distill_16.sh ├── vit-small_QAT_energy.sh ├── vit-small_SNN_16.sh └── vit-small_SNN_16_energy.sh ├── spike_quan_layer.py ├── spike_quan_wrapper.py ├── spikezip_logo.png ├── submitit_finetune.py ├── submitit_linprobe.py ├── submitit_pretrain.py ├── util ├── __pycache__ │ ├── datasets.cpython-310.pyc │ ├── datasets.cpython-39.pyc │ ├── lr_decay.cpython-310.pyc │ ├── lr_decay.cpython-39.pyc │ ├── lr_sched.cpython-310.pyc │ ├── lr_sched.cpython-39.pyc │ ├── misc.cpython-310.pyc │ ├── misc.cpython-39.pyc │ ├── pos_embed.cpython-310.pyc │ └── pos_embed.cpython-39.pyc ├── augment.py ├── crop.py ├── datasets.py ├── lars.py ├── lr_decay.py ├── lr_sched.py ├── misc.py ├── neuromorphic_datasets.py └── pos_embed.py └── vision_transformer.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/SNNWrapper.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/SNNWrapper.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/engine_finetune.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/engine_finetune.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/engine_finetune.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/engine_finetune.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/engine_finetune_snn.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/engine_finetune_snn.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/models_vit.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/models_vit.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/models_vit.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/models_vit.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/spike_quan_layer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/spike_quan_layer.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/spike_quan_layer.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/spike_quan_layer.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/spike_quan_layer_snn.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/spike_quan_layer_snn.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/spike_quan_wrapper.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/spike_quan_wrapper.cpython-310.pyc -------------------------------------------------------------------------------- /__pycache__/spike_quan_wrapper.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/spike_quan_wrapper.cpython-39.pyc -------------------------------------------------------------------------------- /__pycache__/spike_quan_wrapper_ICML.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/__pycache__/spike_quan_wrapper_ICML.cpython-39.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/README.md -------------------------------------------------------------------------------- /energy_consumption_calculation/__init__.py: -------------------------------------------------------------------------------- 1 | from .flops_counter import get_model_complexity_info 2 | -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/engine.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/engine.cpython-310.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/engine.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/engine.cpython-39.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/flops_counter.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/flops_counter.cpython-310.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/flops_counter.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/flops_counter.cpython-39.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/ops.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/ops.cpython-310.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/ops.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/ops.cpython-39.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/utils.cpython-310.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/__pycache__/utils.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/__pycache__/utils.cpython-39.pyc -------------------------------------------------------------------------------- /energy_consumption_calculation/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/engine.py -------------------------------------------------------------------------------- /energy_consumption_calculation/flops_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/flops_counter.py -------------------------------------------------------------------------------- /energy_consumption_calculation/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/ops.py -------------------------------------------------------------------------------- /energy_consumption_calculation/test_spikingformer-8-512_eg.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/test_spikingformer-8-512_eg.log -------------------------------------------------------------------------------- /energy_consumption_calculation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/energy_consumption_calculation/utils.py -------------------------------------------------------------------------------- /engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/engine_finetune.py -------------------------------------------------------------------------------- /engine_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/engine_pretrain.py -------------------------------------------------------------------------------- /extract_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/extract_imagenet.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/main_finetune.py -------------------------------------------------------------------------------- /main_finetune_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/main_finetune_distill.py -------------------------------------------------------------------------------- /main_finetune_distill_dvs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/main_finetune_distill_dvs.py -------------------------------------------------------------------------------- /main_finetune_dvs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/main_finetune_dvs.py -------------------------------------------------------------------------------- /main_finetune_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/main_finetune_raw.py -------------------------------------------------------------------------------- /main_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/main_linprobe.py -------------------------------------------------------------------------------- /main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/main_pretrain.py -------------------------------------------------------------------------------- /metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/metric.py -------------------------------------------------------------------------------- /misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/misc.py -------------------------------------------------------------------------------- /models_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/models_mae.py -------------------------------------------------------------------------------- /models_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/models_vit.py -------------------------------------------------------------------------------- /results/vit-large-imagenet-q32-83.86.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/results/vit-large-imagenet-q32-83.86.txt -------------------------------------------------------------------------------- /results/vit-small-imagenet-q32-81.59.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/results/vit-small-imagenet-q32-81.59.txt -------------------------------------------------------------------------------- /scripts/vit-base_QAT_distill_32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/scripts/vit-base_QAT_distill_32.sh -------------------------------------------------------------------------------- /scripts/vit-large_QAT_distill_32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/scripts/vit-large_QAT_distill_32.sh -------------------------------------------------------------------------------- /scripts/vit-small_QAT_distill_16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/scripts/vit-small_QAT_distill_16.sh -------------------------------------------------------------------------------- /scripts/vit-small_QAT_energy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/scripts/vit-small_QAT_energy.sh -------------------------------------------------------------------------------- /scripts/vit-small_SNN_16.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/scripts/vit-small_SNN_16.sh -------------------------------------------------------------------------------- /scripts/vit-small_SNN_16_energy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/scripts/vit-small_SNN_16_energy.sh -------------------------------------------------------------------------------- /spike_quan_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/spike_quan_layer.py -------------------------------------------------------------------------------- /spike_quan_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/spike_quan_wrapper.py -------------------------------------------------------------------------------- /spikezip_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/spikezip_logo.png -------------------------------------------------------------------------------- /submitit_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/submitit_finetune.py -------------------------------------------------------------------------------- /submitit_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/submitit_linprobe.py -------------------------------------------------------------------------------- /submitit_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/submitit_pretrain.py -------------------------------------------------------------------------------- /util/__pycache__/datasets.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/datasets.cpython-310.pyc -------------------------------------------------------------------------------- /util/__pycache__/datasets.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/datasets.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/lr_decay.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/lr_decay.cpython-310.pyc -------------------------------------------------------------------------------- /util/__pycache__/lr_decay.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/lr_decay.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/lr_sched.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/lr_sched.cpython-310.pyc -------------------------------------------------------------------------------- /util/__pycache__/lr_sched.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/lr_sched.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/misc.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/misc.cpython-310.pyc -------------------------------------------------------------------------------- /util/__pycache__/misc.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/misc.cpython-39.pyc -------------------------------------------------------------------------------- /util/__pycache__/pos_embed.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/pos_embed.cpython-310.pyc -------------------------------------------------------------------------------- /util/__pycache__/pos_embed.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/__pycache__/pos_embed.cpython-39.pyc -------------------------------------------------------------------------------- /util/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/augment.py -------------------------------------------------------------------------------- /util/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/crop.py -------------------------------------------------------------------------------- /util/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/datasets.py -------------------------------------------------------------------------------- /util/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/lars.py -------------------------------------------------------------------------------- /util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/lr_decay.py -------------------------------------------------------------------------------- /util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/lr_sched.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/neuromorphic_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/neuromorphic_datasets.py -------------------------------------------------------------------------------- /util/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/util/pos_embed.py -------------------------------------------------------------------------------- /vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Intelligent-Computing-Research-Group/SpikeZIP-TF/HEAD/vision_transformer.py --------------------------------------------------------------------------------