Upload combined_train.py with huggingface_hub
Browse files- combined_train.py +1 -1
combined_train.py
CHANGED
|
@@ -601,7 +601,7 @@ def train(args):
|
|
| 601 |
print(f"Device: {device}")
|
| 602 |
if device.type == "cuda":
|
| 603 |
print(f"GPU: {torch.cuda.get_device_name()}")
|
| 604 |
-
print(f"VRAM: {torch.cuda.get_device_properties(0).
|
| 605 |
|
| 606 |
# ── Tokenizer ──
|
| 607 |
tokenizer_path = args.tokenizer_path
|
|
|
|
| 601 |
print(f"Device: {device}")
|
| 602 |
if device.type == "cuda":
|
| 603 |
print(f"GPU: {torch.cuda.get_device_name()}")
|
| 604 |
+
print(f"VRAM: {torch.cuda.get_device_properties(0).total_memory / 1e9:.1f} GB")
|
| 605 |
|
| 606 |
# ── Tokenizer ──
|
| 607 |
tokenizer_path = args.tokenizer_path
|