popov/gogo2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix emojies

Browse Source
This commit is contained in:
Dobromir Popov
2025-11-22 21:46:35 +02:00
parent e541e16e7e
commit 9a06288911
37 changed files with 528 additions and 181 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test_model_size.py
View File

@@ -30,14 +30,14 @@ print()
# Check GPU availability
if torch.cuda.is_available():
print(f"GPU Available: ✅ CUDA")
print(f"GPU Available: CUDA")
print(f" Device: {torch.cuda.get_device_name(0)}")
print(f" Memory: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.2f} GB")
# Move model to GPU
device = torch.device('cuda')
model = model.to(device)
print(f" Model moved to GPU ✅")
print(f" Model moved to GPU")
# Test forward pass
batch_size = 1
@@ -50,18 +50,18 @@ if torch.cuda.is_available():
with torch.no_grad():
outputs = model(price_data_1m=price_data_1m)
print(f" Forward pass successful ✅")
print(f" Forward pass successful")
print(f" GPU memory allocated: {torch.cuda.memory_allocated() / 1024**2:.2f} MB")
print(f" GPU memory reserved: {torch.cuda.memory_reserved() / 1024**2:.2f} MB")
elif hasattr(torch.version, 'hip') and torch.version.hip:
print(f"GPU Available: ✅ ROCm/HIP")
print(f"GPU Available: ROCm/HIP")
device = torch.device('cuda') # ROCm uses 'cuda' device name
model = model.to(device)
print(f" Model moved to GPU ✅")
print(f" Model moved to GPU")
else:
print(f"GPU Available: ❌ CPU only")
print(f"GPU Available: CPU only")
print(f" Training will use CPU (slower)")
print()
print("Model ready for training! 🚀")
print("Model ready for training!")