Hi, I would like to extend your answer. Suppose,
running_loss += loss.item()
epoch_loss = running_loss / len(dataset).
I think this is also a way to calculate the loss. We don’t have problem with the batches. Am i right?