From 6f883db1a1bff8655e3d4e5d239c99467af88ce8 Mon Sep 17 00:00:00 2001 From: Nic Ma Date: Wed, 1 Sep 2021 06:41:14 +0800 Subject: [PATCH] [DLMED] simplify random seed Signed-off-by: Nic Ma --- monai/utils/misc.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/monai/utils/misc.py b/monai/utils/misc.py index 3b287b3fe4..a31452f6ae 100644 --- a/monai/utils/misc.py +++ b/monai/utils/misc.py @@ -235,8 +235,7 @@ def set_determinism( if seed is None: # cast to 32 bit seed for CUDA seed_ = torch.default_generator.seed() % (np.iinfo(np.int32).max + 1) - if not torch.cuda._is_in_bad_fork(): - torch.cuda.manual_seed_all(seed_) + torch.manual_seed(seed_) else: seed = int(seed) % MAX_SEED torch.manual_seed(seed)