Swin transformer官方代码只记录了每个iter的损失值，如何才能显示每个epoch的损失值

您可以在训练的代码中添加一个用于记录每个epoch损失值的变量，并在每个epoch结束时将其记录下来。具体来说，您可以添加一个列表losses，并在每个epoch循环结束时，将该epoch的总损失值添加到losses列表中。然后，您可以将该列表中的值绘制成一个损失值随时间变化的折线图。这样，您可以清楚地查看每个epoch的损失值，并对模型的训练效果进行评估。