# A new characterization of the edge of stability based on a sharpness measure aware of batch gradient distribution

This repository is the official implementation of "A new characterization of the edge of stability based on a sharpness measure aware of batch gradient distribution".



## Tutorial

To train a model with GD/SGD, run the following commands:

```training
python main.py
python main_SGD.py
```

See `./codes/run/plot.ipynb` for plotting some main figures in the paper.
