From 454f4d579d7ac06283d62fb9a17e10f97d8c086a Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Sat, 15 Feb 2025 15:25:53 +0800 Subject: [PATCH] Fix bytes to GB conversion for logged VRAM usage (#19254) Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Co-authored-by: UltralyticsAssistant --- ultralytics/engine/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/trainer.py b/ultralytics/engine/trainer.py index dbf0e3cf..81b0c55d 100644 --- a/ultralytics/engine/trainer.py +++ b/ultralytics/engine/trainer.py @@ -493,7 +493,7 @@ class BaseTrainer: memory = 0 else: memory = torch.cuda.memory_reserved() - return memory / 1e9 + return memory >> (2**30) def _clear_memory(self): """Clear accelerator memory on different platforms."""