How can I use class_weights when training?

🤗Transformers
2

Answering my own question:
Subclass Trainer and override the compute_loss method (see example here).

6 Likes
Handling class imbalance when finetuning a decoder model on text generation