-
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
Enable swinunetr-v2 #6203
Enable swinunetr-v2 #6203
Conversation
Signed-off-by: heyufan1995 <[email protected]>
0bb01de
to
f4b11bc
Compare
for more information, see https://pre-commit.ci
Signed-off-by: Wenqi Li <[email protected]>
/black |
Signed-off-by: monai-bot <[email protected]>
The V2 implementation is great, one more thing to discuss here. |
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.
thanks for confirming @tangy5, I'm merging this for now, and I think for future extensions we should start new classes instead of adding more logic to this class...
/build |
/build |
Fixes Project-MONAI#6183 . ### Description Added a "use_v2" option in swinunetr initialization. Default is false will not affect the original swinunetr. Once changed to true, will become swinunetr-v2 with 4 additional convolution block. Tested running from auto3dseg bundles, no change needed for original swinunetr, and works for swinunetr-v2 Tested running from monai research contribution repo for swinuntr, no change needed for original swinunetr, and works for swinunetr-v2 Tested TensorRT, compiled .ts file successfully. ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. --------- Signed-off-by: heyufan1995 <[email protected]>
Fixes Project-MONAI#6183 . ### Description Added a "use_v2" option in swinunetr initialization. Default is false will not affect the original swinunetr. Once changed to true, will become swinunetr-v2 with 4 additional convolution block. Tested running from auto3dseg bundles, no change needed for original swinunetr, and works for swinunetr-v2 Tested running from monai research contribution repo for swinuntr, no change needed for original swinunetr, and works for swinunetr-v2 Tested TensorRT, compiled .ts file successfully. ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. --------- Signed-off-by: heyufan1995 <[email protected]>
Fixes #6183 .
Description
Added a "use_v2" option in swinunetr initialization. Default is false will not affect the original swinunetr.
Once changed to true, will become swinunetr-v2 with 4 additional convolution block.
Tested running from auto3dseg bundles, no change needed for original swinunetr, and works for swinunetr-v2
Tested running from monai research contribution repo for swinuntr, no change needed for original swinunetr, and works for swinunetr-v2
Tested TensorRT, compiled .ts file successfully.
Types of changes
./runtests.sh -f -u --net --coverage
../runtests.sh --quick --unittests --disttests
.make html
command in thedocs/
folder.