mlfoundations / model-soups

Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference time

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Possible to provide minimal code for Figure 2?

sayakpaul opened this issue · comments

Thanks for the awesome paper!

Is it possible to provide minimal code for reproducing some parts of Figure 2? I believe that would greatly benefit the community of researchers that are working on similar things.

As we say in the fig caption, we follow the great work of Garipov et al., 2018 in creating this figure. Their repository is here: https://github.com/timgaripov/dnn-mode-connectivity with plane.py and plane_plot.py being the important files.

Thanks.