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

subclass TrainerCallback (docs 54) to create a custom callback that logs the training metrics by triggering an event with on_evaluate

subclass Trainer and override the evaluate function (docs 56) to inject the additional evaluation code

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