[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

545 stars 46 forks source link

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

Closed aryamanarora closed 4 months ago

aryamanarora commented 4 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