Make tensor smaller to fit in 7gb of gpu memory

n-shevko · n-shevko · commit 717ee17796fb · 2025-04-15T10:34:15.000-04:00
diff --git a/examples/benchmark/sparse_vs_dense_tensors.py b/examples/benchmark/sparse_vs_dense_tensors.py
@@ -12,7 +12,7 @@
 
 assert torch.cuda.is_available(), 'Benchmark works only on cuda'
 device = torch.device("cuda")
-shape = (500, 500, 500)
+shape = (300, 500, 500)
 
 
 def create_spikes_tensor(percent_of_true_values, sparse, return_memory_usage=True):
