Added activation sparsity to the obj detection example.
Makes the following changes:
updates spikingjelly modules with nn.ReLU instead of F.relu
changes NeuroBenchModel.activation_layers() to check for activation module before recursing down the grandchildren, as the spikingjelly neuron model has a nested submodule (surrogate function)
moved snn.SpikingNeuron as recognized activation module to SNNTorchModel rather than NeuroBenchModel
Added activation sparsity to the obj detection example. Makes the following changes:
NeuroBenchModel.activation_layers()
to check for activation module before recursing down the grandchildren, as the spikingjelly neuron model has a nested submodule (surrogate function)snn.SpikingNeuron
as recognized activation module toSNNTorchModel
rather thanNeuroBenchModel