criterion是什么损失函数

criterion是PyTorch中的一个损失函数，用于计算模型预测输出与真实标签之间的差异。常用的criterion包括交叉熵损失函数、均方误差损失函数等。在分类任务中，通常使用交叉熵损失函数；在回归任务中，通常使用均方误差损失函数。通过最小化criterion的值，可以使得模型的预测输出尽可能接近真实标签，达到更好的训练效果。