update batch size

AI4EPS · Sep 16, 2024 · fa2585f · fa2585f
1 parent 63456e2
commit fa2585f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/adloc/data.py b/adloc/data.py
@@ -202,7 +202,7 @@ def __init__(
         picks,
         events,
         stations,
-        batch_size=1000,
+        batch_size=10000,
         valid_index=None,
         config=None,
         rank=0,