runningloss 即计算一个epoch的损失,最后查看的指标是 runningloss/样本数
一些简单的案例
for i in range(epochs):
running_loss=0
for data in dataloader:
....
....
loss.backward()
optimizer.step()
running_loss += loss.item()
print(...)
for i in range(epochs):
....
....
loss.backward()
optimizer.step()
print(loss.item())
第一种情况有将一个epoch的数据划分成batch,所以每次criterion误差函数评估出的loss 需要求和。
第二种情况则是数据集没有划分成batch,所以直接从loss里面取出损失值就行。