about the activation function

[CHonChou]

Is it possible to replace the activation function RELU in snn.py with Prelu

[CHonChou]

In other words, I am talking about the activation function in vgg_spiking.py

[nitin-rathi]

The activation function in vgg_spiking.py is a place holder. The actual activation is integrate-and-fire (IF) implemented in the LinearSpike/STDB class

[CHonChou]

If I change the activation function used by ann from relu to Prelu and avgpool2d to maxpool2d, what should I change in vgg_spiking.py accordingly

[CHonChou]

If I apply the batch_nromalization method in ann, do I need to make corresponding changes in snn

[nitin-rathi]

This ANN-SNN conversion method only works for ReLU, average pooling, and dropout. If you plan to include batch norm, you may need to define additional blocks for SNN.

[CHonChou]

thank you a lot for your replay,i'll try to figure out this.