├── .gitignore ├── LICENSE ├── README.md ├── conf ├── cifar10-dvs │ └── 2_256_200E_t10_TET.yml ├── cifar10 │ ├── 2_256_300E_t4.yml │ ├── 2_512_300E_t4.yml │ └── 2_512_300E_t4_TET.yml ├── cifar100 │ ├── 2_256_300E_t4.yml │ └── 2_512_300E_t4.yml ├── gesture │ ├── 2_256_200E_t10.yml │ └── 2_256_200E_t16.yml └── imagenet │ ├── 10_384_300E_t4.yml │ ├── 6_512_300E_t4.yml │ ├── 8_384_300E_t4.yml │ ├── 8_512_300E_t4.yml │ └── 8_768_300E_t4.yml ├── criterion.py ├── dvs_utils ├── __init__.py ├── cifar10_dvs_loader.py ├── dvs_augment.py └── dvs_utils.py ├── firing_num.py ├── imgs ├── Fig_1_main_idea.png ├── Fig_2_network_architecture.png └── Fig_3_attention_map.png ├── model ├── __init__.py └── spikeformer.py ├── module ├── __init__.py ├── ms_conv.py └── sps.py ├── train.py ├── train.sh └── valid.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/README.md -------------------------------------------------------------------------------- /conf/cifar10-dvs/2_256_200E_t10_TET.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/cifar10-dvs/2_256_200E_t10_TET.yml -------------------------------------------------------------------------------- /conf/cifar10/2_256_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/cifar10/2_256_300E_t4.yml -------------------------------------------------------------------------------- /conf/cifar10/2_512_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/cifar10/2_512_300E_t4.yml -------------------------------------------------------------------------------- /conf/cifar10/2_512_300E_t4_TET.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/cifar10/2_512_300E_t4_TET.yml -------------------------------------------------------------------------------- /conf/cifar100/2_256_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/cifar100/2_256_300E_t4.yml -------------------------------------------------------------------------------- /conf/cifar100/2_512_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/cifar100/2_512_300E_t4.yml -------------------------------------------------------------------------------- /conf/gesture/2_256_200E_t10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/gesture/2_256_200E_t10.yml -------------------------------------------------------------------------------- /conf/gesture/2_256_200E_t16.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/gesture/2_256_200E_t16.yml -------------------------------------------------------------------------------- /conf/imagenet/10_384_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/imagenet/10_384_300E_t4.yml -------------------------------------------------------------------------------- /conf/imagenet/6_512_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/imagenet/6_512_300E_t4.yml -------------------------------------------------------------------------------- /conf/imagenet/8_384_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/imagenet/8_384_300E_t4.yml -------------------------------------------------------------------------------- /conf/imagenet/8_512_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/imagenet/8_512_300E_t4.yml -------------------------------------------------------------------------------- /conf/imagenet/8_768_300E_t4.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/conf/imagenet/8_768_300E_t4.yml -------------------------------------------------------------------------------- /criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/criterion.py -------------------------------------------------------------------------------- /dvs_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/dvs_utils/__init__.py -------------------------------------------------------------------------------- /dvs_utils/cifar10_dvs_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/dvs_utils/cifar10_dvs_loader.py -------------------------------------------------------------------------------- /dvs_utils/dvs_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/dvs_utils/dvs_augment.py -------------------------------------------------------------------------------- /dvs_utils/dvs_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/dvs_utils/dvs_utils.py -------------------------------------------------------------------------------- /firing_num.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/firing_num.py -------------------------------------------------------------------------------- /imgs/Fig_1_main_idea.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/imgs/Fig_1_main_idea.png -------------------------------------------------------------------------------- /imgs/Fig_2_network_architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/imgs/Fig_2_network_architecture.png -------------------------------------------------------------------------------- /imgs/Fig_3_attention_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/imgs/Fig_3_attention_map.png -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | from .spikeformer import sdt 2 | 3 | __all__ = ["sdt"] 4 | -------------------------------------------------------------------------------- /model/spikeformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/model/spikeformer.py -------------------------------------------------------------------------------- /module/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/module/__init__.py -------------------------------------------------------------------------------- /module/ms_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/module/ms_conv.py -------------------------------------------------------------------------------- /module/sps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/module/sps.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/train.sh -------------------------------------------------------------------------------- /valid.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BICLab/Spike-Driven-Transformer/HEAD/valid.sh --------------------------------------------------------------------------------