-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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
Init fp16 support #139
Init fp16 support #139
Conversation
Codecov Report
@@ Coverage Diff @@
## master #139 +/- ##
==========================================
- Coverage 82.32% 82.25% -0.08%
==========================================
Files 143 145 +2
Lines 6621 6711 +90
Branches 986 994 +8
==========================================
+ Hits 5451 5520 +69
- Misses 1071 1090 +19
- Partials 99 101 +2
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
# support fp16 | ||
self.fp16_enabled = False | ||
|
||
@auto_fp16(apply_to=('x', )) |
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.
@auto_fp16()
should be enough
# support fp16 | ||
self.fp16_enabled = False | ||
|
||
@auto_fp16(apply_to=('x', 'mask_in')) |
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.
here
# support fp16 | ||
self.fp16_enabled = False | ||
|
||
@auto_fp16(apply_to=('x', )) |
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.
here
mmedit/models/mattors/dim.py
Outdated
# support fp16 | ||
self.fp16_enabled = False | ||
|
||
@auto_fp16(apply_to=('x', 'refine')) |
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.
here
* unfinished stage * support fp16 * fix comment from ck * fix comment from ck * fix comment from ck Co-authored-by: nbei <[email protected]>
* unfinished stage * support fp16 * fix comment from ck * fix comment from ck * fix comment from ck Co-authored-by: nbei <[email protected]>
No description provided.