Hao Li, Zheng Xu, Gavin Taylor, Christoph Studer and Tom Goldstein. Visualizing the Loss Landscape of Neural Nets. NIPS, 2018. An interactive 3D visualizer for loss surfaces has been provided by ...
@jnolck thanks for your reply. can you give me some information about your setup ? what is your cpu/gpu ? are you on Linux Ubuntu ? what are your pytorch and python versions ? what is the rocm version ...