Pytorch nn.不指定目标的MSELoss

Question

我的损失在一个特定的值上卡住了。它总是会减小到某个值，然后停止减小。关于损失的代码是:

criterion = nn.MSELoss()
loss = criterion(y_pred, y_batch.unsqueeze(1))

当我把它改成:

criterion = nn.MSELoss()
loss = criterion(y_pred, target=y_batch)

问题已修复。

在没有指定目标之前发生了什么?是否需要为每个Pytorch损失函数指定目标?我在文档中没有发现任何关于目标规格的信息。

Answer 1

看起来target是第二个位置参数的名称，仅此而已。这两条线之间唯一的区别是dim=1在第二条线上的解压缩。