PyLandscape

Introduction

pylandscape is a Pytorch library for loss landscape analysis of neural networks. The library enables computing the following metrics:

• CKA similarity
• Hessian metrics
• Mode connectivity
• Loss surface

NOTE: All the functionalities relative to the computation of the Hessian metrics have been embedded via PyHessian. If your interested in learning more about how these metrics are computed have a look to their Repository.

Usage

Install from Pip

You can install the library from pip:

pip install pylandscape