Fix train.py: mem -> memory on line 225

scripts/train.py

@@ -222,7 +222,7 @@ def main() -> None:
     print(f"  Device:      {'cuda' if torch.cuda.is_available() else 'cpu'}")
     if torch.cuda.is_available():
         print(f"  GPU:         {torch.cuda.get_device_name(0)}")
-        vram_gb = torch.cuda.get_device_properties(0).total_mem / (1024 ** 3)
+        vram_gb = torch.cuda.get_device_properties(0).total_memory / (1024 ** 3)
         print(f"  VRAM:        {vram_gb:.1f} GB")
     print()