@phan_phan thanks for the reply!
I went ahead and modified my code to accumulate the averages of the values as you suggested. My results look a little better but I think maybe more so just confirms the problem further haha
For reference my:
starting training loss was 0.016 and validation was 0.0019,
final training loss was 0.004 and validation loss was 0.0007.
And here’s a viz of the losses over ten epochs of training. Based on this, I think the model is improving and I’m not calculating validation loss correctly, but I can’t figure out anything I’m doing wrong!