khanrc / swad

Official Implementation of SWAD (NeurIPS 2021)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Loss surface plots

JeanKaddour opened this issue · comments

Hi,

Thank you so much for providing this repo, the code looks great.

How can one reproduce the loss surface plots, e.g., Figure 4 in the paper?

Did you use this code here?

Hi,

We wrote our own code to plot Figure 4, referring to the code you linked. I think you can use that code since both implementations work on the same principle.

Thank you very much! :)