gather_neurons() unit tests

stanfordnlp / pyvene

Stanford NLP Python Library for Understanding and Improving PyTorch Models via Interventions

http://pyvene.ai

Apache License 2.0

559 stars 50 forks source link

Closed ZhengPeterWang closed 6 months ago

ZhengPeterWang commented 6 months ago

Only tested h, pos, and h.pos cases for now. Also added some doc string to gather_neurons()