Better docstring. Image size and device found in forward. #12

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.

Sign up for GitHub

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

Jump to bottom

Merged

ndrplz merged 1 commit into ndrplz:master from arroqc:init_tweaks

Feb 27, 2020

arroqc commented Feb 24, 2020

Mods:

Removed Variable calls (old pytorch version)
Removed dependency on image size (now inferred by the batch size)
Clarified output and docstring
Fixed device hardcoded for hidden states
Some PEP8 stuff


          Better docstring. Image size and device found in forward.

6d61754

ndrplz merged commit 3607865 into ndrplz:master

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

Labels

None yet