Add pytorch-vision #1599
Open
Add pytorch-vision #1599
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Module for adding the ability to run on pytorch nn models that use operation implemented on torch vision.
|
This looks like something that we could add to the presets for PyTorch? |
|
You are suggesting to be included in the normal build of pytorch presets? Yes I can do that. |
|
Yes, that's easier to maintain, but it looks like this only works on Linux? There is no support for Mac or Windows? Let me think about what we should do with this... |
|
I didn't test on other OS. Because of that, I didn't want to add code that
I wasn't sure it worked.
Hopping on a windows and/or Mac developer to add it later on :)
|
|
Right, we can add that as a Linux-only module to the presets for PyTorch for now, that's fine I think |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Module for adding the ability to run on pytorch nn models that use operation implemented on torch vision.