Figure 5: Loss surface using [35]; SGD (top) and PSGD (bottom)