From a02eee46438532d98fa9b68e841aaf769a96d98c Mon Sep 17 00:00:00 2001
From: Charles Gaydon <charles.gaydon@gmail.com>
Date: Mon, 16 Oct 2023 13:29:44 +0200
Subject: [PATCH] prefetch_factor=1

---
 .github/workflows/staging-model.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/staging-model.yml b/.github/workflows/staging-model.yml
index 6b005004..1e77fc1e 100644
--- a/.github/workflows/staging-model.yml
+++ b/.github/workflows/staging-model.yml
@@ -35,6 +35,7 @@ jobs:
         predict.src_las=/data/val/*.las
         predict.output_dir=/lidar_prod_optimization/${model_id}/preds-valset/
         predict.interpolator.probas_to_save=[building]
+        datamodule.prefetch_factor=1
 
     - name: Inference on test set of 20211001_building_val-lidar-prod-optimization
       run: >
@@ -49,6 +50,7 @@ jobs:
         predict.src_las=/data/test/*.las
         predict.output_dir=/lidar_prod_optimization/${model_id}/preds-testset/
         predict.interpolator.probas_to_save=[building]
+        datamodule.prefetch_factor=1
 
     # NB: we run the new, tagged image at least once so that is it not prunned
     # by mistake when using docker system prune.