Project-MONAI / MONAI

AI Toolkit for Healthcare Imaging
https://monai.io/
Apache License 2.0
5.87k stars 1.09k forks source link

NACLLoss memory management #8020

Closed Bala93 closed 3 months ago

Bala93 commented 3 months ago

Fixes # .

Description

Calling contiguous after applying the permute option to work with view operation in apply_filter (https://github.com/Project-MONAI/MONAI/blob/59a7211070538586369afd4a01eca0a7fe2e742e/monai/networks/layers/simplelayers.py#L293).

Types of changes

KumoLiu commented 3 months ago

Hi @Bala93, please help sign off to fix the DCO issue, then I can help trigger more tests. https://github.com/Project-MONAI/MONAI/pull/8020/checks?check_run_id=28797284498

Bala93 commented 3 months ago

Hi @KumoLiu, thanks for the immediate response. I also added the test case for testing with batch size>1, which was failing in the previous version.

KumoLiu commented 3 months ago

/build