fix extract data error in rgcn

uwsampl · Feb 15, 2023 · 39ec694 · 39ec694
1 parent a2fe42b
commit 39ec694
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rgcn/extract_data.py b/rgcn/extract_data.py
@@ -47,7 +47,7 @@ def extract_data():
                 mem_str = display_name
                 assert (dataset, "graphiler") in exec_times
                 graphiler_dur = exec_times[(dataset, "graphiler")]
-                graphiler_mem = exec_times[(dataset, "graphiler")]
+                graphiler_mem = mem[(dataset, "graphiler")]
 
                 if not (dataset, "pyg-bmm") in mem and not (dataset, "pyg-slice") in mem:
                     speed_str += " 0"