fix: total_memory attribute name

run_training.py

@@ -24,7 +24,7 @@ def run_grpo_training():
 
     if torch.cuda.is_available():
         print(f"GPU: {torch.cuda.get_device_name(0)}")
-        print(f"VRAM: {torch.cuda.get_device_properties(0).total_mem / 1e9:.1f} GB")
+        print(f"VRAM: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB")
     else:
         print("WARNING: No GPU detected — training will be very slow!")