pytorch - 关于将什么输入到损失函数中的问题

Question

def fit(num_epochs, model, loss_fn, opt, train_dl):
    for epoch in range(num_epochs):
        for xb,yb in train_dl:
            pred = model(xb)
            loss = loss_fn(pred, yb)

train_dl只是由 DataLoader 拆分成合适批次的数据集。Model()是激活函数。无论如何，为什么使用 yb 而不是 xb 计算损失？这是一个计算损失的教程代码，我认为代码应该看起来像这样。

def fit(num_epochs, model, loss_fn, opt, train_dl):
    for epoch in range(num_epochs):
        for xb in train_dl:
            pred = model(xb)
            loss = loss_fn(pred, xb)

教程来自这个网站。https://jovian.ai/aakashns/02-linear-regression

标签： pytorch