-
Notifications
You must be signed in to change notification settings - Fork 26.3k
bug fix 19374 - fix for upsample export #20116
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
Conversation
|
bug fix #19374 |
250dfe6 to
d7faf52
Compare
houseroad
left a comment
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.
Looks good, can you add a test case to cover the code you add?
Here is an example: https://github.com/pytorch/pytorch/blob/master/test/onnx/test_pytorch_onnx_caffe2.py#L958
d7faf52 to
5688c7e
Compare
Sure, thanks for the reference. Added the test to cover new code. |
facebook-github-bot
left a comment
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.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
5688c7e to
d21c1e7
Compare
facebook-github-bot
left a comment
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.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
We should enable the test for Upsample in test_operators.py also. pytorch/test/onnx/test_operators.py Line 476 in 4d676d5
|
|
torch/onnx/symbolic.py
Outdated
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.
This shouldn't be deleted
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.
I guess there has been updates to upsample that for the argument output_size it could be either a tensor or a int list. Thus the annotation becomes @parse_args('v', 'v') which can be ignored.
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.
That's right, as @BowenBao mentioned adding @pars_args('v','v') will be redundant.
houseroad
left a comment
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.
See inline comments
d21c1e7 to
4c75ed8
Compare
@Alcaster Good point, we should definitely address other methods as well. But since in this PR we are trying to fix and unblock issue #19374 (which uses 'nearest' method), how about addressing other methods in a separate PR ? |
|
The failing onnx ci is legit. Please fix the failing tests, thanks. |
|
Does this also fix the "bilinear" upsampling export? |
facebook-github-bot
left a comment
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.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@pk-g friendly ping for fix :-) |
|
Pls fix =) |
4c75ed8 to
60ae121
Compare
60ae121 to
1867cd0
Compare
facebook-github-bot
left a comment
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.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@pk-g please rebase again to make sure the CI pass. The broken CI has been fixed already. Thanks! |
1867cd0 to
d7010eb
Compare
facebook-github-bot
left a comment
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.
@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
houseroad
left a comment
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.
LGTM, thanks for fixing the problem.
|
@houseroad merged this pull request in 93d5503. |
No description provided.