Code based off https://github.com/BICLab/Spike-Driven-Transformer

Modified for multi-level LIF integration and training in module/mlif.py

An ImageNet dataset is assumed to be prepared in ./data

For training M-LIF SNNs on CIFAR10, CIFAR100 or ImageNet, see train.py
For estimating energy using checkpoint, see firing_num.py