Swin transformer目标检测官方代码只记录了每个iter的损失值,如何才能显示每个epoch的损失值
时间: 2023-06-09 17:02:36 浏览: 135
基于swin-transformer实现小目标检测任务python源码+使用说明+数据集.zip
要显示每个epoch的损失值,你需要在代码中添加一个或多个变量来记录每个epoch的损失值,并且在epoch结束后将其显示在屏幕上或保存到文件中。可以将每个epoch的损失值定义为一个列表或数组,每次迭代都将损失值添加到该列表或数组中,然后在epoch结束时,计算该列表或数组的平均值,并将其显示在屏幕上或保存到文件中。还要确保你在记录损失值时,使用的是在epoch级别上的数据,而不是每个iter上的数据。
阅读全文