Skip to content

2499 adds 2d/3d support of patchembedding, ViT, and unetr model #2698

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.

Sign up for GitHub

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

Jump to bottom
Merged
merged 8 commits into from
Aug 6, 2021
Merged

2499 adds 2d/3d support of patchembedding, ViT, and unetr model #2698

merged 8 commits into from
Aug 6, 2021

Conversation

wyli
Copy link
Contributor

@wyli wyli commented Aug 5, 2021 •
edited
Loading

Signed-off-by: Wenqi Li [email protected]

part of #2499

Description

the current patchembedding/vit model only supports 3d input,
this PR extends them for 2d

Status

Ready

Types of changes

  • Non-breaking change (fix or new feature that would not break existing functionality).
  • New tests added to cover the changes.
  • Quick tests passed locally by running ./runtests.sh --quick --unittests.
  • In-line docstrings updated.
  • Documentation updated, tested make html command in the docs/ folder.

@wyli wyli requested review from ahatamiz and Nic-Ma August 5, 2021 16:07
@wyli
2d/3d patchembedding
363e574 
Signed-off-by: Wenqi Li <[email protected]>
@wyli wyli force-pushed the 2499-2d-patchembedding branch from d2722a7 to 363e574 Compare August 5, 2021 16:10
@wyli wyli changed the title adds support of 2d patchembedding adds support of 2d/3d patchembedding Aug 5, 2021
@ahatamiz
Copy link
Contributor

ahatamiz commented Aug 5, 2021 •
edited by wyli
Loading

/black

@wyli wyli changed the title adds support of 2d/3d patchembedding adds support of 2d/3d patchembedding and the ViT model Aug 5, 2021
@wyli
update unetr
f822f67 
Signed-off-by: Wenqi Li <[email protected]>
@wyli wyli changed the title adds support of 2d/3d patchembedding and the ViT model adds 2d/3d support of patchembedding, ViT, and unetr model Aug 5, 2021
@wyli wyli changed the title adds 2d/3d support of patchembedding, ViT, and unetr model 2499 adds 2d/3d support of patchembedding, ViT, and unetr model Aug 5, 2021
@wyli
fixes unit test
10d6117 
Signed-off-by: Wenqi Li <[email protected]>
Nic-Ma
Nic-Ma approved these changes Aug 6, 2021
View reviewed changes
Copy link
Contributor

@Nic-Ma Nic-Ma left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the enhancement.
PR looks good to me.
@ahatamiz could you please help double confirm as you are expert on the UNETR, I may missed to check some computation theory.

Thanks in advance.

ahatamiz
ahatamiz approved these changes Aug 6, 2021
View reviewed changes
Copy link
Contributor

@ahatamiz ahatamiz left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @wyli for the great work.

This PR looks good to me.

@ahatamiz ahatamiz merged commit 62425d7 into Project-MONAI:dev Aug 6, 2021
@wyli wyli deleted the 2499-2d-patchembedding branch August 6, 2021 14:20
@ahatamiz
Copy link
Contributor

ahatamiz commented Aug 6, 2021

Thanks for the enhancement.
PR looks good to me.
@ahatamiz could you please help double confirm as you are expert on the UNETR, I may missed to check some computation theory.

Thanks in advance.

Thanks @Nic-Ma for your review.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ahatamiz ahatamiz ahatamiz approved these changes

@Nic-Ma Nic-Ma Nic-Ma approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@wyli @ahatamiz @Nic-Ma