## Requirements

The following list of libraries is required:

`torch`, `torchvision`, `numpy`, `tqdm`, `three_regimes_on_the_sphere` (https://github.com/tipt0p/three_regimes_on_the_sphere)

## Running code

1. Run the script `run_fourier_accs.py`. You need to specify `lrs` --- a list of floats, denoting the PLR values.
Also set the dict `ckpts_by_lr`, which uses PLRs as key and stores a list of paths to the checkpoints, corresponding to PLRs.
If you are interested in different architectures/datasets, change the respective lines of code.
2. Use the notebook `plots-final.ipynb` for plotting figures.