This repository has been archived by the owner on Aug 18, 2021. It is now read-only.

small format issue #151

Open

jinghuayao opened this issue Mar 31, 2021 · 0 comments

jinghuayao commented Mar 31, 2021 •

edited

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

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.