From 686f9b530b14bfab4eea5488533d92fa6725b33b Mon Sep 17 00:00:00 2001 From: Sherlock Date: Tue, 27 Jul 2021 15:43:13 -0700 Subject: [PATCH] ORTModule set_seed in int (#8511) --- orttraining/orttraining/python/training/ortmodule/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/orttraining/orttraining/python/training/ortmodule/__init__.py b/orttraining/orttraining/python/training/ortmodule/__init__.py index 4d87131abc..d86eb4adeb 100644 --- a/orttraining/orttraining/python/training/ortmodule/__init__.py +++ b/orttraining/orttraining/python/training/ortmodule/__init__.py @@ -51,13 +51,13 @@ set_seed((torch.initial_seed() % sys.maxsize)) # Override torch.manual_seed and torch.cuda.manual_seed def override_torch_manual_seed(seed): - set_seed(seed % sys.maxsize) + set_seed(int(seed % sys.maxsize)) return torch_manual_seed(seed) torch_manual_seed = torch.manual_seed torch.manual_seed = override_torch_manual_seed def override_torch_cuda_manual_seed(seed): - set_seed(seed % sys.maxsize) + set_seed(int(seed % sys.maxsize)) return torch_cuda_manual_seed(seed) torch_cuda_manual_seed = torch.cuda.manual_seed torch.cuda.manual_seed = override_torch_cuda_manual_seed