How to get Loss landscape
ShenZheng2000 opened this issue · 2 comments
ShenZheng2000 commented
Hello authors! I really appreciate your excellent work. For Fig. 4 (the Loss landscape) in your paper, I want to know the following information:
(1) The plotting code script, package, or repository
(2) The testing dataset
(3) The model version
Thanks!
Shen Zheng
ma-xu commented
Hi @ShenZheng2000 Thanks for your interest.
- 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.
- The testing dataset is ModelNet40, as indicated in Sec. Loss landscape.
- We test PointMLP and PointMLP without skip connections.
Please let me know if you have any further questions.
ShenZheng2000 commented
Thanks for your timely reply. Got it now.