archinetai / surgeon-pytorch

A library to inspect and extract intermediate layers of PyTorch models.
470Updated 2 years ago

Related projects

Alternatives and complementary repositories for surgeon-pytorch