/surgeon-pytorch

A library to inspect and extract intermediate layers of PyTorch models.

Primary LanguagePythonMIT LicenseMIT

Watchers