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

2
ultralytics/engine/trainer.py
View file

@ -52,6 +52,7 @@ from ultralytics.utils.torch_utils import (
select_device,
strip_optimizer,
torch_distributed_zero_first,
unset_deterministic,
)
@ -471,6 +472,7 @@ class BaseTrainer:
self.plot_metrics()
self.run_callbacks("on_train_end")
self._clear_memory()
unset_deterministic()
self.run_callbacks("teardown")
def auto_batch(self, max_num_obj=0):