update cctorch

AI4EPS · Nov 24, 2023 · 4c759bb · 4c759bb
1 parent f773778
commit 4c759bb
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/slurm/run_cctorch.py b/slurm/run_cctorch.py
@@ -28,13 +28,10 @@ def parse_args():
 
 
 ## based on GPU memory
-# batch = 2048
-# block_size1 = 1000
-# block_size2 = 1000
 
-batch = 100
-block_size1 = 10
-block_size2 = 10
+batch = 1_024
+block_size1 = 10_000
+block_size2 = 10_000  # ~7GB
 
 if args.dtct_pair:
     dt_ct = f"{root_path}/{region}/hypodd/dt.ct"