Is there anyone who can give me some idea how i can plot a loss landscape like below in pytorch?
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!