32mb buffer

recursionpharma · Feb 23, 2024 · 648961f · 648961f
1 parent 9c83ce5
commit 648961f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gflownet/tasks/seh_frag.py b/src/gflownet/tasks/seh_frag.py
@@ -193,7 +193,7 @@ def main():
         "overwrite_existing_exp": True,
         "num_training_steps": 100,
         "num_workers": 8,
-        "mp_buffer_size": None,# 32 * 1024 ** 2,
+        "mp_buffer_size": 32 * 1024 ** 2,
         "opt": {
             "lr_decay": 20000,
         },