ndif-team / nnsight

The nnsight package enables interpreting and manipulating the internals of deep learned models.
https://nnsight.net/
MIT License
356 stars 34 forks source link

When doing global patching on a class vs a fn need to #219

Closed JadenFiotto-Kaufman closed 1 week ago

JadenFiotto-Kaufman commented 1 week ago

1.) Patch the init method not the class itself to preserve type checking like isinstance 2.) Still make sure the nnsight tracing fn applied is the class not the init method