Skip to content
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

Auto3DSeg ensemble needs to take images from different devices #6361

Closed
mingxin-zheng opened this issue Apr 14, 2023 · 1 comment
Closed
Labels
bug Something isn't working

Comments

@mingxin-zheng
Copy link
Contributor

Describe the bug
Referece link:
Project-MONAI/research-contributions#211

Different algorithms may return results on different devices (CPU or GPU) , e.g. if algo didn't have enough gpu memory it may do CPU based slidingwindow inference. So than when we do Ensembling, some internal function concatinate the predictions and it fails, since the predictions are on different devices.

@wyli wyli added the bug Something isn't working label Apr 14, 2023
@wyli wyli added this to MONAI v1.2 Apr 14, 2023
wyli pushed a commit that referenced this issue Apr 15, 2023
Fixes # #6361

Ensures ensemble preds are on the same device

---------

Signed-off-by: myron <amyronenko@nvidia.com>
@wyli
Copy link
Contributor

wyli commented Apr 19, 2023

Addressed by #6368

@wyli wyli closed this as completed Apr 19, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
No open projects
Status: Done
Development

No branches or pull requests

2 participants