Skip to content
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

#902-Patch avoid moving model #910

Merged
merged 6 commits into from Apr 10, 2020
Merged

Conversation

kai-tub
Copy link
Contributor

@kai-tub kai-tub commented Apr 9, 2020

Fixes #902

Description:

  • Remove the model move
  • Update the docstring and include a warning note
  • Updated the tests to make sure the behavior is as expected

Grepped the repo and made the required changes to keep everything
in-line with the new changes to create_supervised.
Hopefully, I didn't miss anything.
I did not rerun the FastaiLRFinder_MINST.ipynb.
The output would change and add unnecessary size to the repo.

Check list:

  • New tests are added (if a new feature is added)
  • New doc strings: description and/or example code are in RST format
  • Documentation is updated (if required)

k.clasen and others added 3 commits April 9, 2020 22:06
- Remove the model move
- Update the docstring and include a warning note
- Updated the tests to make sure the behavior is as expected.
Grepped the repo and made the required changes to keep everything
in-line with the new changes to `create_supervised`.

Hopefully I didn't miss anything.
I did *not* rerun the FastaiLRFinder_MINST.ipynb.
The output would change and add unnecessary size to the rep.
Copy link
Collaborator

@vfdev-5 vfdev-5 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM! @kai-tub thanks !

@vfdev-5 vfdev-5 merged commit 400c26a into pytorch:master Apr 10, 2020
Minyus added a commit to Minyus/pipelinex that referenced this pull request Apr 12, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Avoid moving model to device in create_supervised*
2 participants