This message was deleted.

Hi @marah othman. Do you mean the evolution of the loss during training?

yes

For creating plots manually, models store the train (and validation loss) for each step in the

train_trajectories

object. You can access it after training and plot the evolution (easiest and recommended). Alternatively, you can use PyTorch Lightning automatic logs and integration with TensorBoard.

okay thanks i will see