Changed persistent_workers=True to False

datamol-io · Sep 19, 2024 · 24412f6 · 24412f6
1 parent f723632
commit 24412f6
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/expts/hydra-configs/architecture/largemix.yaml b/expts/hydra-configs/architecture/largemix.yaml
@@ -85,7 +85,7 @@ datamodule:
   args:
     processed_graph_data_path: ${constants.datacache_path}
     num_workers: 20 # -1 to use all
-    persistent_workers: True
+    persistent_workers: False
     featurization:
       atom_property_list_onehot: [atomic-number, group, period, total-valence]
       atom_property_list_float: [degree, formal-charge, radical-electron, aromatic, in-ring]

diff --git a/expts/hydra-configs/finetuning/admet_baseline.yaml b/expts/hydra-configs/finetuning/admet_baseline.yaml
@@ -20,7 +20,7 @@ constants:
 datamodule:
   args:
     batch_size_training: 32
-    persistent_workers: true
+    persistent_workers: false
     num_workers: 4
 
 trainer:

diff --git a/expts/neurips2023_configs/base_config/large.yaml b/expts/neurips2023_configs/base_config/large.yaml
@@ -164,7 +164,7 @@ datamodule:
             ksteps: 16
 
     num_workers: 32 # -1 to use all
-    persistent_workers: True # if use persistent worker at the start of each epoch.
+    persistent_workers: False # if use persistent worker at the start of each epoch.
     # Using persistent_workers false might make the start of each epoch very long.
 
 

diff --git a/expts/neurips2023_configs/base_config/large_pcba.yaml b/expts/neurips2023_configs/base_config/large_pcba.yaml
@@ -163,7 +163,7 @@ datamodule:
             ksteps: 16
 
     num_workers: 32 # -1 to use all
-    persistent_workers: True # if use persistent worker at the start of each epoch.
+    persistent_workers: False # if use persistent worker at the start of each epoch.
     # Using persistent_workers false might make the start of each epoch very long.
 
 

diff --git a/expts/neurips2023_configs/base_config/large_pcqm_g25.yaml b/expts/neurips2023_configs/base_config/large_pcqm_g25.yaml
@@ -163,7 +163,7 @@ datamodule:
             ksteps: 16
 
     num_workers: 32 # -1 to use all
-    persistent_workers: True # if use persistent worker at the start of each epoch.
+    persistent_workers: False # if use persistent worker at the start of each epoch.
     # Using persistent_workers false might make the start of each epoch very long.
 
 

diff --git a/expts/neurips2023_configs/base_config/large_pcqm_n4.yaml b/expts/neurips2023_configs/base_config/large_pcqm_n4.yaml
@@ -163,7 +163,7 @@ datamodule:
             ksteps: 16
 
     num_workers: 32 # -1 to use all
-    persistent_workers: True # if use persistent worker at the start of each epoch.
+    persistent_workers: False # if use persistent worker at the start of each epoch.
     # Using persistent_workers false might make the start of each epoch very long.