Effective Transfer Learning Algorithm in Spiking Neural Networks.
Python implementation of transfer learning on SNNs with the centered kernel alignment (CKA).
Requirement
- Python 3.7
- Pytorch 1.7.1+cu101
- prefetch_generator 1.0.1
- tqdm 4.54.1
Training and testing
To run the codes on transfer learning dataset in /transfer_data, you should first download the corrosponding dataset file. Then, you should put the samples into different folders as each folder requires (e.g., /transfer_data/PACS/giraffe/image_names.txt).
To do so, you can run the following command to train and test the model:
$ python snn_tl.py