PyTorch implementation of the empirical results described in "The Implicit Bias of Gradient Descent on Separable Data" (ICLR-2018) by Daniel Soudry, Elad Hoffer, Mor Shpigel Nacson, Suriya Gunasekar, Nathan Srebro.
@inproceedings{
soudry2018the,
title={The Implicit Bias of Gradient Descent on Separable Data},
author={Daniel Soudry and Elad Hoffer and Nathan Srebro},
booktitle={International Conference on Learning Representations},
year={2018},
url={https://openreview.net/forum?id=r1q7n9gAb},
}
To get the relevant figures simply run
python main.py --save maxmargin_results
- pytorch
- torchvision to load the datasets, perform image transforms
- pandas for logging to csv
- bokeh for training visualization
- Configure your dataset path at data.py.