zhuzhucheng

摘要： aver_loss 总损失的计算 对于求平均损失来说 需要先求总损失 而求总损失 就需要求一个批次中的损失 对于一个bs来说 损失的计算是利用 loss=criterion(out,labels)计算得出 而criterion 使用的nn.crossentropy 得出来的损失值 已经是对这一个bs 阅读全文

摘要： 适用于在pytorch的张量上，求某一维度的最大值。 一般在模型测试阶段，求模型预测输出类别的时候使用。 假设是10分类问题，比如mnist 对于一个批次的输入 images 将它传入net(images) 会得到输出out(bs,10) 但是第二个维度，仅仅是模型在十个类别的预测值 需要取最大值， 阅读全文