摘要:
aver_loss 总损失的计算 对于求平均损失来说 需要先求总损失 而求总损失 就需要求一个批次中的损失 对于一个bs来说 损失的计算是利用 loss=criterion(out,labels)计算得出 而criterion 使用的nn.crossentropy 得出来的损失值 已经是对这一个bs 阅读全文
posted @ 2025-09-24 13:08
朱朱成
阅读(28)
评论(0)
推荐(0)
摘要:
适用于在pytorch的张量上,求某一维度的最大值。 一般在模型测试阶段,求模型预测输出类别的时候使用。 假设是10分类问题,比如mnist 对于一个批次的输入 images 将它传入net(images) 会得到输出out(bs,10) 但是第二个维度,仅仅是模型在十个类别的预测值 需要取最大值, 阅读全文
posted @ 2025-09-24 13:00
朱朱成
阅读(8)
评论(0)
推荐(0)

浙公网安备 33010602011771号