search

hunkim / PyTorchZeroToAll

Simple PyTorch Tutorials Zero to ALL!
http://bit.ly/PyTorchZeroAll
3.9k stars 1.2k forks source link

Fixed 05_linear_regression.py #47

Open akihiro-inui opened 5 years ago

akihiro-inui commented 5 years ago

To extract loss, loss.data[0] did not work so I removed [0] and it worked.