Add batch_size parameter to inference configuration files

NRCan · May 1, 2024 · 71b60c2 · 71b60c2
1 parent 24dd60b
commit 71b60c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/config/inference/default_binary.yaml b/config/inference/default_binary.yaml
@@ -6,6 +6,7 @@ inference:
   model_path: ${general.save_weights_dir}/
   output_path:
   checkpoint_dir:  # (string, optional): directory in which to save the object if url
+  batch_size: 8
   chunk_size:  # if empty, will be calculated automatically from max_pix_per_mb_gpu
   # Maximum number of pixels each Mb of GPU Ram to allow. E.g. if GPU has 1000 Mb of Ram and this parameter is set to
   # 10, chunk_size will be set to sqrt(1000 * 10) = 100.

diff --git a/config/inference/default_multiclass.yaml b/config/inference/default_multiclass.yaml
@@ -6,6 +6,7 @@ inference:
   model_path: ${general.save_weights_dir}/
   output_path:
   checkpoint_dir:  # (string, optional): directory in which to save the object if url
+  batch_size: 8
   chunk_size:  # if empty, will be calculated automatically from max_pix_per_mb_gpu
   # Maximum number of pixels each Mb of GPU Ram to allow. E.g. if GPU has 1000 Mb of Ram and this parameter is set to
   # 10, chunk_size will be set to sqrt(1000 * 10) = 100.