Thank you for your reply. I can already plot the loss curve for one epoch, but I was looking to plot the loss for a number of epochs. Using the manual approach you suggested above, this is the loss curve generated:

What I would like to do is something like this:

image.png812×612 43.6 KB

I hope that clears up what I wanting to do.