Skip to content
This repository was archived by the owner on Aug 18, 2021. It is now read-only.

Conversation

PatrikHolopainen
Copy link

Changed the train() -function so that they work on PyTorch 1.3.1

Changed target[c] --> target[c].view(1) (because of dimensionality problems)
Changed loss.data[0] --> loss.data.item()

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant