The currently supported modules in the snntorch's _exportnir.py are [Leaky, Linear, Synaptic, RLeaky, RSynaptic]. In my case I need to convert an SNN model with Conv2D, Leaky, Linear and AveragePooling2D layers to NIR.
What could be done
Two possible options to solve this:
Naively add these two torch.nn layer operations to the _snntorch/exportnir.py function
Centralizing NIRTorch functions _fromnir.py and _tonir.py to support the NIR conversion of torch.nn layers regardless of the used framework (snntorch,norse,sinabs...)
Description
The currently supported modules in the snntorch's _exportnir.py are [Leaky, Linear, Synaptic, RLeaky, RSynaptic]. In my case I need to convert an SNN model with Conv2D, Leaky, Linear and AveragePooling2D layers to NIR.
What could be done
Two possible options to solve this: