From c9a65fb00e8bf4206934e6486885045c4650dcd8 Mon Sep 17 00:00:00 2001 From: Charles Gaydon Date: Mon, 16 Oct 2023 09:58:37 +0200 Subject: [PATCH] fix: batch_size=2 --- .github/workflows/staging-model.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/staging-model.yml b/.github/workflows/staging-model.yml index 54b177cb..a02cd77c 100644 --- 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.