05_linear_regression

[ys7yoo]

/usr/local/lib/python3.6/dist-packages/torch/nn/_reduction.py:49: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
  warnings.warn(warning.format(ret))
---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-2-f85695c5ca83> in <module>()
     38     # Compute and print loss
     39     loss = criterion(y_pred, y_data)
---> 40     print(epoch, loss.data[0])
     41 
     42     # Zero gradients, perform a backward pass, and update the weights.

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

[ys7yoo]

fixed print(epoch, loss.data[0]) to print(epoch, loss.item())

[ys7yoo]

Error is gone.
But, still have a warning

/usr/local/lib/python3.6/dist-packages/torch/nn/_reduction.py:49: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
  warnings.warn(warning.format(ret))

[ys7yoo]

replaced

criterion = torch.nn.MSELoss(size_average=False)

with

criterion = torch.nn.MSELoss(reduction='sum')