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

4
docs/en/reference/utils/torch_utils.md
View file

@ -119,6 +119,10 @@ keywords: Ultralytics, torch utils, model optimization, device selection, infere
<br><br><hr><br>
## ::: ultralytics.utils.torch_utils.unset_deterministic
<br><br><hr><br>
## ::: ultralytics.utils.torch_utils.strip_optimizer
<br><br><hr><br>