diff --git a/ultralytics/engine/trainer.py b/ultralytics/engine/trainer.py
index dbf0e3cfea..81b0c55df2 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."""