Question about how to extract loss for each epoch?

[Joanne-PYL]

Hi,

Thank you for the amazing work. I am using the model training cells trajectories. I believe that the result I trained is overfitting. I am relatively new to deep learning algorithms. I am wondering that is there any way to extract training and validation loss for each epoch, so that I can adjust the parameters and layer numbers in the model? Thank you for your help.

Best,
Joanne

[Joanne-PYL]

Never mind, I found the solution haha.