From 4d30f143371c8c4102bd94a837428e587556663a Mon Sep 17 00:00:00 2001
From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Date: Mon, 17 Feb 2025 14:32:09 +0800
Subject: [PATCH] Fix memory conversion (#19256)

Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
---
 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 81b0c55df2..5e7173b2a8 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 >> (2**30)
+        return memory / (2**30)
 
     def _clear_memory(self):
         """Clear accelerator memory on different platforms."""