How to monitor both train and validation metrics at the same step?

hey @Mariam, as far as i know you’ll have to create your own Trainer to evaluate the metrics on the training set - see this thread for a related discussion: Logging training accuracy using Trainer class