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

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

View file

@ -140,6 +140,7 @@ class Model(torch.nn.Module):
return
# Load or create new YOLO model
__import__("os").environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8" # to avoid deterministic warnings
if Path(model).suffix in {".yaml", ".yml"}:
self._new(model, task=task, verbose=verbose)
else: