fix nb workers

kermitt2 · Nov 6, 2024 · 9292e2a · 9292e2a
1 parent 0d7151a
commit 9292e2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/delft/sequenceLabelling/trainer.py b/delft/sequenceLabelling/trainer.py
@@ -193,7 +193,7 @@ def train_model(self, local_model, x_train, y_train, f_train=None,
                 model=local_model,
                 external_callbacks=callbacks
             )
-        nb_workers = 0
+        nb_workers = 6
         multiprocessing = self.training_config.multiprocessing
 
         # multiple workers should work with transformer layers, but not with ELMo due to GPU memory limit (with GTX 1080Ti 11GB)