ma-xu/pointMLP-pytorch

How to visual Fig. 4 in your paper

Closed this issue · 4 comments

Thank you for your amazing work!
Could you share the code of drawing Fig.4 (Loss landscape along two rand directions.)

ma-xu commented

I will send it to you by the end of May (or early June). Sry that I'm really busy recently.

I also have the same request. Can you share the code?

@Elbenhorn @LiuHahahaha Sorry about the late reply.

We firstly get the raw data by lossland.py, then we plot the figure by plot.py. Note that both scripts are not cleaned andn you may need to change some settiings accordingly (e.g., checkpoint paths, ranges, etc.). More details can be found here: Li, Hao, et al. "Visualizing the loss landscape of neural nets." NeurIPS 2018.

Also in issue: #42

Very appreciate it!