|
|
@ -319,9 +319,9 @@ def init_seeds(seed=0, deterministic=False):
|
|
|
|
torch.cuda.manual_seed(seed)
|
|
|
|
torch.cuda.manual_seed(seed)
|
|
|
|
torch.cuda.manual_seed_all(seed) # for Multi-GPU, exception safe
|
|
|
|
torch.cuda.manual_seed_all(seed) # for Multi-GPU, exception safe
|
|
|
|
# torch.backends.cudnn.benchmark = True # AutoBatch problem https://github.com/ultralytics/yolov5/issues/9287
|
|
|
|
# torch.backends.cudnn.benchmark = True # AutoBatch problem https://github.com/ultralytics/yolov5/issues/9287
|
|
|
|
if deterministic: # https://github.com/ultralytics/yolov5/pull/8213
|
|
|
|
if deterministic:
|
|
|
|
if TORCH_2_0:
|
|
|
|
if TORCH_2_0:
|
|
|
|
torch.use_deterministic_algorithms(True)
|
|
|
|
torch.use_deterministic_algorithms(True, warn_only=True) # warn if deterministic is not possible
|
|
|
|
torch.backends.cudnn.deterministic = True
|
|
|
|
torch.backends.cudnn.deterministic = True
|
|
|
|
os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
|
|
|
|
os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
|
|
|
|
os.environ['PYTHONHASHSEED'] = str(seed)
|
|
|
|
os.environ['PYTHONHASHSEED'] = str(seed)
|
|
|
|