Skip to content

rtao499/shenjingcat

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

40 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ShenjingCat: An SNN framework with PyTorch

Authors: Jun Zhou, Zhanglu Yan

Installation

The repository includes C++ and CUDA code that has to be compiled and installed before it can be used from Python, download the repository and run the following command to do so:

python setup.py install

Documentation

Please refer to the wiki

Training Workflow

CQ Trainig is trained in stages as the figure below. The details of training methods is in learning_tricks.md.

Publications:

Z. Yan, J. Zhou, and W.F. Wong, "Near Lossless Transfer Learning for Spiking Neural Networks." Accepted by Thirty-Fifth AAAI Conference on Artificial Intelligence (AAAI-21). Virtual conference. Feb 2021.

Acknowledgement

This simulator is inspired by and makes use of the code from slayerPytorch

License & Copyright

Copyright 2020 Jun Zhou, ShenjingCat is free software: you can redistribute it and/or modoify it under the terms of GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

ShenjingCat is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details http://www.gnu.org/licenses/.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 95.6%
  • C++ 3.1%
  • Cuda 1.2%
  • Shell 0.1%