[P1] Add `IntervenableModel.named_parameters()` - Githubissues

stanfordnlp / pyvene

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

http://pyvene.ai

Apache License 2.0

652 stars 63 forks source link

[P1] Add `IntervenableModel.named_parameters()` #122

Closed aryamanarora closed 9 months ago

aryamanarora commented 9 months ago

Description

This adds the named_parameters() method which is used by HF trainer to find the trainable parameters to pass to the optimiser (#121).

Testing Done

No effect on any existing code.

Checklist:

[x] My PR title strictly follows the format: [Your Priority] Your Title
[x] I have attached the testing log above
[x] I provide enough comments to my code
[x] I have changed documentations
[x] I have added tests for my changes