search

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

[Minor] Accepting `labels` field for loss calculation #136

Closed frankaging closed 3 months ago

frankaging commented 3 months ago

Description

Letting the forward API to accept the labels field.

Testing Done

Unit tests.

Checklist: