Code to reproduce the results from Neurally Augmented ALISTA, ICLR 2021. Freya Behrens, Jonathan Sauder, Peter Jung.
Experiment demo:
python run.py
for parameterization, refer to the comments in the file.
Reproduce the figures from the paper:
jupyter notebook plot_results.ipynb
Experiment demo:
python run-communication.py
for parameterization, refer to the comments in the file.
Reproduce the figures from the paper:
jupyter notebook plot_results_communication.ipynb
- Without a GPU the experiments take a lot of time.
- We provide some precomputed results in the
res/
folder.