-
Notifications
You must be signed in to change notification settings - Fork 7k
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
Make Googlnet & InceptionNet scriptable #1349
Conversation
blocked by pytorch/pytorch#26437 |
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!
4d10bab
to
46dd08c
Compare
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, except for the test, which has a spurious assert now
test/test_models.py
Outdated
@@ -55,7 +53,7 @@ def check_script(self, model, name): | |||
tb = traceback.format_exc() | |||
scriptable = False | |||
msg = str(e) + str(tb) | |||
self.assertEqual(torchub_models[name], scriptable, msg) | |||
self.assertEqual(scriptable, scriptable, msg) |
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'm not sure this change is what you want to do?
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.
Since they are all scriptable now, we don't need to access torchub_models
and just assert that scriptable
is True
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.
Can you then make
self.assertEqual(scriptable, True, msg)
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.
oh oops - good catch
Test failures look legit |
@pytorchbot rebase this please |
…enet_scriptable
…enet_scriptable
Codecov Report
@@ Coverage Diff @@
## master #1349 +/- ##
==========================================
+ Coverage 63.9% 63.94% +0.03%
==========================================
Files 78 78
Lines 6147 6176 +29
Branches 940 944 +4
==========================================
+ Hits 3928 3949 +21
- Misses 1941 1947 +6
- Partials 278 280 +2
Continue to review full report at Codecov.
|
Based on the conclusion from #1273: we are always returning a tuple when scripting.
This was blocked by pytorch/pytorch#26683.