New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

export/import_nir__Adding-Conv2D_and_AvgPool2d #305

Open

SirineArfa wants to merge 3 commits into jeshraghian:master from SirineArfa:Adding-Conv2D_and_Pooling

SirineArfa commented Mar 25, 2024

fixes Missing Conv2D and AveragePooling2D modules in export/import to NIR #304
Adding support for the following torch.nn modules [Conv2d, AvgPool2d and Flatten] to export_nir.py and import_nir.py
Create example notebook in examples/tutorial_snntorch_to_nir.ipynb for snntorch model with layers [Conv2d, AvgPool2d,LIF, Linear,LIF] and export/import it to NIR
Include this model's tests in tests/test_nir.py

SirineArfa added 2 commits

March 25, 2024 00:32


          export/import_nir__Adding-Conv2D_and_AvgPool2d

e523435


          code formatting

5bf33e7

Author

SirineArfa commented Apr 10, 2024

The pip version of NIR is not automatically updated and is currently version 1.0.1 from Dec 25. Hence, it does not contain AvgPool2d yet.
To try this feature you must build NIR from source:
- git clone https://github.com/neuromorphs/NIR.git
- cd NIR
- pip install .


          edit export and import

89a7f3c

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment