Fix memory conversion (#19256)

Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
2025-02-17 14:32:09 +08:00 · 2025-02-17 14:32:09 +08:00 · 4d30f14337
commit 4d30f14337
parent 454f4d579d
1 changed files with 1 additions and 1 deletions
--- 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."""