73260e319a82fbc14e8984962f903838ffdf13dc,1_linear_regression.py,,train,#,12

Before Change

    optimizer.zero_grad()

    // Forward
    fx = model.forward(x.view(len(x), 1))
    output = loss.forward(fx, y)

    // Backward
    output.backward()

    // Update parameters
    optimizer.step()

    return output.data[0]


def main():
    torch.manual_seed(42)

After Change

    optimizer.zero_grad()

    // Forward
    fx = model.forward(x.view(len(x), 1)).squeeze()
    output = loss.forward(fx, y)

    // Backward
    output.backward()

Instances