From 213b9dc32d8ff220fba0f33475d71e114f9cf4af Mon Sep 17 00:00:00 2001 From: KoalaTrapLord Date: Sun, 5 Apr 2026 18:06:50 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20total=5Fmem=20=E2=86=92=20total=5Fmemory?= =?UTF-8?q?=20in=20trainer.py=20(PyTorch=20API)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ml/training/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml/training/trainer.py b/ml/training/trainer.py index 9de19c4..a0dd8b3 100644 --- a/ml/training/trainer.py +++ b/ml/training/trainer.py @@ -547,7 +547,7 @@ def get_best_device() -> str: return "cpu" if torch.cuda.is_available(): name = torch.cuda.get_device_name(0) - mem = torch.cuda.get_device_properties(0).total_mem / (1024**3) + mem = torch.cuda.get_device_properties(0).total_memory / (1024**3) logger.info(" CUDA device: %s (%.1f GB)", name, mem) return "cuda" if hasattr(torch.backends, "mps") and torch.backends.mps.is_available():