pytorch AverageMeter

最新推荐文章于 2024-02-20 21:33:45 发布

tony365

最新推荐文章于 2024-02-20 21:33:45 发布

阅读量260

点赞数

分类专栏： pytorch 文章标签： pytorch python 人工智能

本文链接：https://blog.csdn.net/tywwwww/article/details/131080722

版权

pytorch 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

AverageMeter类用于计算和存储平均值和当前值。在每个训练epoch开始时初始化，每个迭代(iter)中更新损失值，使用输入样本的数量进行加权。to_python_float函数确保转换为Python浮点数以保持向后兼容性。

摘要由CSDN通过智能技术生成

平均量

class AverageMeter(object):
    """Computes and stores the average and current value"""
    def __init__(self):
        self.reset()

    def reset(self):
        self.val = 0
        self.avg = 0
        self.sum = 0
        self.count = 0

    def update(self, val, n=1):
        self.val = val
        self.sum += val * n
        self.count += n
        self.avg = self.sum / self.count

调用：
训练，每个epoch前：
losses = AverageMeter()

当前epoch 每个iter:

losses.update(to_python_float(reduced_loss.data), input.size(0))

# item() is a recent addition, so this helps with backward compatibility.
def to_python_float(t):
    if hasattr(t, 'item'):
        return t.item()
    else:
        return t[0]