Logging training accuracy using Trainer class

🤗Transformers
2

hey @dbejarano31, assuming that you want to log the training metrics during training, i think there are (at least) two options:

  1. subclass TrainerCallback (docs) to create a custom callback that logs the training metrics by triggering an event with on_evaluate
  2. subclass Trainer and override the evaluate function (docs) to inject the additional evaluation code

option 2 might be easier to implement since you can use the existing logic as a template :slight_smile:

3 Likes