HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Unset CUBLAS_WORKSPACE_CONFIG for non-deterministic training and inference (#19138)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Mohammed Yasin 2025-02-10 07:17:08 +08:00 • committed by GitHub
parent 31e96e20d3
commit ff1a04609f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 16 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
ultralytics/utils/torch_utils.py
View file

@ -488,8 +488,15 @@ def init_seeds(seed=0, deterministic=False):
else:
LOGGER.warning("WARNING ⚠️ Upgrade to torch>=2.0.0 for deterministic training.")
else:
torch.use_deterministic_algorithms(False)
torch.backends.cudnn.deterministic = False
unset_deterministic()
def unset_deterministic():
"""Unsets all the configurations applied for deterministic training."""
torch.use_deterministic_algorithms(False)
torch.backends.cudnn.deterministic = False
os.environ.pop("CUBLAS_WORKSPACE_CONFIG", None)
os.environ.pop("PYTHONHASHSEED", None)
class ModelEMA: