add final runbook

harsha-simhadri · Oct 13, 2023 · b71cf6c · b71cf6c
1 parent d8ddce4
commit b71cf6c
Show file tree

Hide file tree

Showing 4 changed files with 3,853 additions and 15 deletions.
diff --git a/benchmark/datasets.py b/benchmark/datasets.py
@@ -448,11 +448,11 @@ def __init__(self):
         self.d = 100
         self.nq = 10000
         self.dtype = "float32"
-        self.ds_fn = "msturing-29998994-clustered.fbin"
+        self.ds_fn = "30M-clustered64.fbin"
         self.qs_fn = "testQuery10K.fbin"
         self.gt_fn = "clu_msturing30M_gt100"
 
-        self.base_url = ""
+        self.base_url = "https://comp21storage.blob.core.windows.net/publiccontainer/comp23/clustered_data/msturing-30M-clustered/"
         self.basedir = os.path.join(BASEDIR, "MSTuring-30M-clustered")
 
         self.private_gt_url = None

diff --git a/benchmark/runner.py b/benchmark/runner.py
@@ -293,7 +293,7 @@ def run_docker(definition, dataset, count, runs, timeout, rebuild,
     # set/override container timeout based on competition flag
     if neurips23track!='none':
         # 1 hour for streaming and 12 hours for other tracks
-        timeout = 60 * 60 if neurips23track == 'streaming' else 12 * 60 * 60
+        timeout = 2 * 60 * 60 if neurips23track == 'streaming' else 12 * 60 * 60
         print("Setting container wait timeout to %d seconds" % timeout)       
 
     elif not timeout: