Add paralelism and workers for many steps

allenai · Dec 19, 2024 · 1312ea1 · 1312ea1
1 parent 98cbcfb
commit 1312ea1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/rslp/forest_loss_driver/inference/config/forest_loss_driver_predict_pipeline_config.yaml b/rslp/forest_loss_driver/inference/config/forest_loss_driver_predict_pipeline_config.yaml
@@ -1,6 +1,6 @@
 # Required fields
-model_predict_args:
-  model_cfg_fname: "data/forest_loss_driver/config_satlaspretrain_flip_oldmodel_unfreeze.yaml" # should be path from the top of the repo IF NOT ABSOLUTE PATH
+extract_alerts_args:
+  workers: 112
 materialize_pipeline_args:
   prepare_args:
     apply_windows_args:
@@ -11,3 +11,7 @@ materialize_pipeline_args:
   materialize_args:
     apply_windows_args:
       workers: 112
+select_best_images_args:
+  workers: 112
+model_predict_args:
+  model_cfg_fname: "data/forest_loss_driver/config_satlaspretrain_flip_oldmodel_unfreeze.yaml" # should be path from the top of the repo IF NOT ABSOLUTE PATH