How to plot loss landscape in pytorch?

Is there anyone who can give me some idea how i can plot a loss landscape like below in pytorch?

Screen Shot 2021-10-06 at 9.34.05 AM

This type of plot is a surface plot and you could use matplotlib for it. I don’t know what the current recommended technique is to create this loss surface from a DL model, but e.g. this paper might be useful.

1 Like

Looking for the same thing here, have you managed to find any solutions yet? Thanks!