khanrc / swad

Official Implementation of SWAD (NeurIPS 2021)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to draw Figure 4?

WentaoTan opened this issue · comments

commented

This is a beautifully drawn picture!
Can you tell me how to draw this kind of loss value and model parameter correlation diagram?

We release the reference code to help. See https://github.com/khanrc/swad/tree/main/visualization.

commented

I'm sorry to bother you again, but I've looked at the code in the visualization folder and the SWA related code, and I still don't understand it, and I've encountered a lot of errors. The main problems are:
(1) The import file infer_utils is empty.
(2) What is the content of the file with the suffix "*.pth"?

Thanks a lot!

As written in the README, the visualization code is provided for reference only, not for execution. *.pth files indicate checkpoints. This code, which is our original reference code, may help you better.