Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Aviv Navon 2001d52f3b | 8 months ago | |
---|---|---|
experiments | 1 year ago | |
misc | 1 year ago | |
nn | 1 year ago | |
notebooks | 1 year ago | |
tests | 1 year ago | |
.gitignore | 1 year ago | |
.isort.cfg | 1 year ago | |
LICENCE | 1 year ago | |
README.md | 8 months ago | |
requirements.txt | 1 year ago | |
setup.py | 1 year ago |
Official implementation for Equivariant Architectures for Learning in Deep Weight Spaces by Aviv Navon, Aviv Shamsian, Idan Achituve, Ethan Fetaya, Gal Chechik, Haggai Maron.
Our implementation follows the block structure as described in the paper.
To run the experiments, first create clean virtual environment and install the requirements.
conda create -n dwsnets python=3.9
conda activate dwsnets
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
Install the repo:
git clone https://github.com/AvivNavon/DWSNets.git
cd DWSNets
pip install -e .
An introduction notebook for INR classification with DWSNets:
To run specific experiment, please follow the instructions in the README file within each experiment folder.
It provides full instructions and details for downloading the data and reproducing the results reported in the paper.
The datasets are available here.
If you find our work or this code to be useful in your own research, please consider citing the following paper:
@InProceedings{pmlr-v202-navon23a,
title = {Equivariant Architectures for Learning in Deep Weight Spaces},
author = {Navon, Aviv and Shamsian, Aviv and Achituve, Idan and Fetaya, Ethan and Chechik, Gal and Maron, Haggai},
booktitle = {Proceedings of the 40th International Conference on Machine Learning},
pages = {25790--25816},
year = {2023},
editor = {Krause, Andreas and Brunskill, Emma and Cho, Kyunghyun and Engelhardt, Barbara and Sabato, Sivan and Scarlett, Jonathan},
volume = {202},
series = {Proceedings of Machine Learning Research},
month = {23--29 Jul},
publisher = {PMLR},
pdf = {https://proceedings.mlr.press/v202/navon23a/navon23a.pdf},
url = {https://proceedings.mlr.press/v202/navon23a.html},
}
No Description
Jupyter Notebook Python
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》