update num_workers

weakly_supervised_parser/model/data_module_loader.py

@@ -58,7 +58,7 @@ class DataModule(LightningDataModule):
         max_seq_length: int = 256,
         train_batch_size: int = 32,
         eval_batch_size: int = 32,
-        num_workers: int = 16,
+        num_workers: int = 4,
         **kwargs
     ):
         super().__init__()

weakly_supervised_parser/utils/populate_chart.py

@@ -28,7 +28,7 @@ ptb_most_common_first_token = "the"
 
 from pytorch_lightning import Trainer
 
-trainer = Trainer(accelerator="auto", enable_progress_bar=False)
+trainer = Trainer(accelerator="auto", enable_progress_bar=False, max_epochs=-1)
 
 
 class PopulateCKYChart: