Update PharmProfiler.py

Wang-Lin-boop · Oct 31, 2024 · 98b55f9 · 98b55f9
1 parent f7cc536
commit 98b55f9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/geminimol/PharmProfiler.py b/geminimol/PharmProfiler.py
@@ -146,6 +146,7 @@ def __call__(
     model_name = sys.argv[1]
     encoder = GeminiMol(
             model_name,
+            batch_size = 512, 
             internal_label_list = [],
             extrnal_label_list = ['Pearson'],
         )