I want to plot a Loss curve line during my training data process which is to take each epoch value loss record and plot a curve line. I\'ve tried the code below and errors poppe