/Joint-A-SNN

Joint A-SNN: Joint Training of Artificial and Spiking Neural Networks via Self-Distillation and Weight Factorization

Primary LanguagePython

Joint A-SNN

Joint A-SNN: Joint Training of Artificial and Spiking Neural Networks via Self-Distillation and Weight Factorization

For Training, please use command:

python main.py --time (timestep) --epochs (total training epochs) --batch_size (batchsize)