From e6a072302c9a23d2b8b22001b8d32f84cbefaa09 Mon Sep 17 00:00:00 2001 From: Lenz Fiedler Date: Wed, 8 Jan 2025 14:15:28 +0100 Subject: [PATCH] Can I use DDP and MPI at the same time? --- mala/common/parallelizer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mala/common/parallelizer.py b/mala/common/parallelizer.py index e59b8a98..8e394999 100644 --- a/mala/common/parallelizer.py +++ b/mala/common/parallelizer.py @@ -64,10 +64,10 @@ def set_mpi_status(new_value): Value the MPI status has. """ - if use_ddp is True and new_value is True: - raise Exception( - "Cannot use ddp and inference-level MPI at " "the same time yet." - ) + # if use_ddp is True and new_value is True: + # raise Exception( + # "Cannot use ddp and inference-level MPI at " "the same time yet." + # ) global use_mpi use_mpi = new_value if use_mpi: