From 7cd2885726c30cb594faee7000f2098d0eb3e872 Mon Sep 17 00:00:00 2001 From: Kyle Swanson Date: Tue, 21 Feb 2023 15:06:16 -0800 Subject: [PATCH] Updating memory requirements --- tree_search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tree_search.py b/tree_search.py index 45b0a34..e93c1ad 100644 --- a/tree_search.py +++ b/tree_search.py @@ -94,7 +94,7 @@ class Args(Tap): store_nodes: bool = False """Whether to store in memory all the nodes of the search tree. This doubles the speed of the search but significantly increases - the memory usage (e.g., ~450GB for 20,000 rollouts instead of <16GB).""" + the memory usage (e.g., 450 GB for 20,000 rollouts instead of 600 MB).""" class TreeNode: @@ -215,7 +215,7 @@ def __init__(self, :param debug: Whether to print out additional statements for debugging. :param store_nodes: Whether to store the child nodes of each node in the search tree. This doubles the speed of the search but significantly increases - the memory usage (e.g., ~450GB for 20,000 rollouts instead of <16GB). + the memory usage (e.g., 450GB for 20,000 rollouts instead of 600 MB). """ self.search_type = search_type self.fragment_smiles_to_id = fragment_smiles_to_id