-
Notifications
You must be signed in to change notification settings - Fork 2.6k
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
[Fix] Add setr & vit msg. #635
Conversation
@@ -25,6 +25,7 @@ def __init__(self, | |||
num_convs=1, | |||
up_scale=4, | |||
kernel_size=3, | |||
init_cfg=None, |
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.
We can set the default init_cfg
to implement the same weights initialization as that in init_weights
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.
And we can set it to None
in configs if SETR does not use it.
Codecov Report
@@ Coverage Diff @@
## master #635 +/- ##
==========================================
- Coverage 85.78% 85.77% -0.02%
==========================================
Files 103 103
Lines 5312 5307 -5
Branches 859 857 -2
==========================================
- Hits 4557 4552 -5
Misses 583 583
Partials 172 172
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
* [Fix] Add setr & vit msg. * Fix init bug * Modify init_cfg arg * Add conv_seg init
* correcting the beta value assignment * updating DDIM and LMSDiscreteFlax schedulers * bringing back the changes that were lost as part of main branch merge
* fix opencv-python 4.5.2 compatibility BottomUpRandomAffine fails with opencv-python>=4.5.2 because a single valued numpy.ndarray is no longer a valid data type for image_size in cv2.warpAffine. This commit fix it by repacing image_size with image_size.item(). * fix smplx 0.1.27 compatibility smplx>=0.1.27 requires loguru, which will not be automatically installed when installing smplx. So we add loguru into mmpose requirements.
Motivation
The lack of SETR & ViT in the model list.
The SETRUPHead still use deprecated init_weights.
Modification
Add SETR & ViT url in the main readme doc.
Refactor init of SETRUPHead.