Here, we provide the jupyter notebooks of quickly reproducing the small-scale experiments in the paper. 
The code for cifar-10 experiments will be available soon.

- alignment.ipynb 	Report the alignment strength during the SGD training
- escape.ipynb		Visualize the process of escaping from a sharp minimum
- flatness.ipynb 	Examine how the extent of over-parameterization impacts the flatness of solutions found by SGD