search

spro / practical-pytorch

Go to https://github.com/pytorch/tutorials - this repo is deprecated and no longer maintained
MIT License
4.52k stars 1.11k forks source link

small format issue #151

Open jinghuayao opened 3 years ago

jinghuayao commented 3 years ago

In the train.py script, Line 45: "return output, loss.data[0]", ---> "return output, loss.item()" is suggested.