guychuk commited on
Commit
cb89707
·
verified ·
1 Parent(s): ca875e1

Upload combined_train.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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).total_mem / 1e9:.1f} GB")
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