Spaces:
Build error
Build error
update num_workers
Browse files
weakly_supervised_parser/model/data_module_loader.py
CHANGED
@@ -58,7 +58,7 @@ class DataModule(LightningDataModule):
|
|
58 |
max_seq_length: int = 256,
|
59 |
train_batch_size: int = 32,
|
60 |
eval_batch_size: int = 32,
|
61 |
-
num_workers: int =
|
62 |
**kwargs
|
63 |
):
|
64 |
super().__init__()
|
|
|
58 |
max_seq_length: int = 256,
|
59 |
train_batch_size: int = 32,
|
60 |
eval_batch_size: int = 32,
|
61 |
+
num_workers: int = 4,
|
62 |
**kwargs
|
63 |
):
|
64 |
super().__init__()
|
weakly_supervised_parser/utils/populate_chart.py
CHANGED
@@ -28,7 +28,7 @@ ptb_most_common_first_token = "the"
|
|
28 |
|
29 |
from pytorch_lightning import Trainer
|
30 |
|
31 |
-
trainer = Trainer(accelerator="auto", enable_progress_bar=False)
|
32 |
|
33 |
|
34 |
class PopulateCKYChart:
|
|
|
28 |
|
29 |
from pytorch_lightning import Trainer
|
30 |
|
31 |
+
trainer = Trainer(accelerator="auto", enable_progress_bar=False, max_epochs=-1)
|
32 |
|
33 |
|
34 |
class PopulateCKYChart:
|