practical-pytorch icon indicating copy to clipboard operation
practical-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon spro

small format issue

Open jinghuayao opened this issue 4 years ago • 0 comments

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

jinghuayao avatar Mar 31 '21 18:03 jinghuayao