├── .gitignore ├── README.md ├── diagnose.py ├── neurips_poster.pdf ├── src ├── data.py ├── linalg.py ├── models │ ├── mnist.py │ ├── resnet.py │ └── vgg.py ├── trainer.py └── utils.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/README.md -------------------------------------------------------------------------------- /diagnose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/diagnose.py -------------------------------------------------------------------------------- /neurips_poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/neurips_poster.pdf -------------------------------------------------------------------------------- /src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/src/data.py -------------------------------------------------------------------------------- /src/linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/src/linalg.py -------------------------------------------------------------------------------- /src/models/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/src/models/mnist.py -------------------------------------------------------------------------------- /src/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/src/models/resnet.py -------------------------------------------------------------------------------- /src/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/src/models/vgg.py -------------------------------------------------------------------------------- /src/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/src/trainer.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/src/utils.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leiwu0/sgd.stability/HEAD/train.py --------------------------------------------------------------------------------