update config

AllenCellModeling · Sep 20, 2024 · 6998235 · 6998235
1 parent 0c4bdca
commit 6998235
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/experiment/im2im/segmentation_plugin.yaml b/configs/experiment/im2im/segmentation_plugin.yaml
@@ -52,7 +52,7 @@ data:
   cache_dir: ${paths.output_dir}/cache # string path to cache_dir (this speeds up data loading)
   num_workers: MUST_OVERRIDE # this should be set based on the number of available CPU cores
   split_column: null
-  batch_size: 1
+  batch_size: AUTO # this will use the automatic batch finder starting with batch_size=1
   _aux:
     patch_shape: [16, 32, 32]