fix: batch_size=2

IGNF · Oct 16, 2023 · c9a65fb · c9a65fb
1 parent 85da86c
commit c9a65fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/staging-model.yml b/.github/workflows/staging-model.yml
@@ -36,7 +36,7 @@ jobs:
         predict.output_dir=/lidar_prod_optimization/${model_id}/preds-valset/
         predict.interpolator.probas_to_save=[building]
         datamodule.prefetch_factor=1
-        datamodule.batch_size=1
+        datamodule.batch_size=2
 
     - name: Inference on test set of 20211001_building_val-lidar-prod-optimization
       run: >
@@ -52,7 +52,7 @@ jobs:
         predict.output_dir=/lidar_prod_optimization/${model_id}/preds-testset/
         predict.interpolator.probas_to_save=[building]
         datamodule.prefetch_factor=1
-        datamodule.batch_size=1
+        datamodule.batch_size=2
 
     # NB: we run the new, tagged image at least once so that is it not prunned
     # by mistake when using docker system prune.