refactor: Add multiprocessing context to improve performance

ylab-hi · Nov 4, 2024 · e69c335 · e69c335
1 parent d53c716
commit e69c335
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/deepchopper/cli.py b/deepchopper/cli.py
@@ -151,6 +151,10 @@ def predict(
         logger=False,
         limit_predict_batches=limit_predict_batches,
     )
+
+    import multiprocess.context as ctx
+
+    ctx._force_start_method("spawn")
     trainer.predict(model=model, dataloaders=datamodule, return_predictions=False)