You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The currently supported modules in the snntorch's export_nir.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/export_nir.py function
Centralizing NIRTorch functions from_nir.py and to_nir.py to support the NIR conversion of torch.nn layers regardless of the used framework (snntorch,norse,sinabs...)
The text was updated successfully, but these errors were encountered:
Description
The currently supported modules in the snntorch's export_nir.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:
The text was updated successfully, but these errors were encountered: