How to prevent `CUDA out of memory` error for a large MONAI network ...