search

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

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

Closed frankaging closed 8 months ago

frankaging commented 8 months ago

Description

Letting the forward API to accept the labels field.

Testing Done

Unit tests.

Checklist: