Code for temporal efficient training
The Following Setup is tested and it is working:
ZIF
in models/layer.py
for step function forward and backward.Layer
in models/layer.py
. \
self.conv = nn.Sequential(Layer(2,64,3,1,1),Layer(64,128,3,1,1),)
Reference paper.
@inproceedings{
deng2022temporal,
title={Temporal Efficient Training of Spiking Neural Network via Gradient Re-weighting},
author={Shikuang Deng and Yuhang Li and Shanghang Zhang and Shi Gu},
booktitle={International Conference on Learning Representations},
year={2022},
url={https://openreview.net/forum?id=_XNtisL32jv}
}