swa
swa copied to clipboard
Figure 3 plot
Hi, can you update the code for plotting Figure 3 in the paper?
Hey @byan19! We used the code from our other project (repo) to make these visualizations.
Please see plane.py for evaluating the necessary values of the loss and plane_plot.py for making the visualization from these values.
You may be interested in the conversation in this issue https://github.com/timgaripov/dnn-mode-connectivity/issues/9.