restore deterministic init_seeds code

ultralytics · Aug 20, 2022 · 8b70e64 · 8b70e64
1 parent 1a84f47
commit 8b70e64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/segment/train.py b/segment/train.py
@@ -99,7 +99,7 @@ def train(hyp, opt, device):  # hyp is path/to/hyp.yaml or hyp dictionary
     plots = not evolve and not opt.noplots  # create plots
     overlap = opt.overlap_mask
     cuda = device.type != 'cpu'
-    init_seeds(opt.seed + 1 + RANK, True)
+    init_seeds(opt.seed + 1 + RANK, deterministic=False)
     with torch_distributed_zero_first(LOCAL_RANK):
         data_dict = data_dict or check_dataset(data)  # check if None
     train_path, val_path = data_dict['train'], data_dict['val']

diff --git a/utils/general.py b/utils/general.py
@@ -242,7 +242,7 @@ def init_seeds(seed=0, deterministic=False):
     import torch.backends.cudnn as cudnn
 
     if deterministic and check_version(torch.__version__, '1.12.0'):  # https://github.com/ultralytics/yolov5/pull/8213
-        # torch.use_deterministic_algorithms(True)
+        torch.use_deterministic_algorithms(True)
         os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
         os.environ['PYTHONHASHSEED'] = str(seed)