Visualizing Generalization Phenomena
The code for this paper is split up into different directories for each section of the paper, and each directory has its own README for how to run the code.
minefieldcontains the code to train a good model and look for bad minima along its trajectory (Figure 1)
linear_modelcontains the code for training the overparametrized linear model and neural net (Figure 2)
optimizerscontains the code for training on CIFAR using different optimizers (Figure 2, Table 1)
swissrollcontains the code for all figures on the swiss roll dataset (Figures 3, 5, 6)
svhncontains the code for all figures on the SVHN dataset (Figures 4, 5, 7)
concentric_circlescontains the code for the concentric circles wide margins experiments (Figure 8)