07_diabets_logistic.py

[ys7yoo]

/anaconda3/envs/torch/lib/python3.5/site-packages/torch/nn/_reduction.py:49: UserWarning: size_average and reduce args will be deprecated, please use reduction='mean' instead.
  warnings.warn(warning.format(ret))
Traceback (most recent call last):
  File "07_diabets_logistic.py", line 55, in <module>
    print(epoch, loss.data[0])
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

[ys7yoo]

replaced

criterion = torch.nn.BCELoss(size_average=True)

with

criterion = torch.nn.BCELoss(reduction='mean')

[ys7yoo]

replaced

    print(epoch, loss.data[0])

with

    print(epoch, loss.item())