-
Notifications
You must be signed in to change notification settings - Fork 8
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fixed implementation of pytorch cuda device #32
Conversation
Passed pytorch_device argument to sleepnet
Fixed for implementation of pytorch use of cuda device
@LeandroCasiraghi thanks a lot for reporting this issue and making a PR. |
There seems to be some format issues about the PR, which seems to be easy to change. Would you mind taking a look quickly :D @LeandroCasiraghi |
fixed format issues
fixed format issues
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed 5 format issues from tests
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just a final formatting issue
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hopefully final formatting issue
We discovered the pytorch_device argument of get_sleep was not being passed on to sleepnet and hence never used.
We also fixed the sleepnet code to correctly implement the use of a cuda device in pytorch and added a printed message that indicates the device being used, either the default 'cpu' or the cuda device input in the get_sleep call.