From 6998235c8ba913106d8f79725f9b2cca780d4d14 Mon Sep 17 00:00:00 2001 From: Benjamin Morris Date: Fri, 20 Sep 2024 12:52:45 -0700 Subject: [PATCH] update config --- configs/experiment/im2im/segmentation_plugin.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/experiment/im2im/segmentation_plugin.yaml b/configs/experiment/im2im/segmentation_plugin.yaml index f5fd632b..49b6ae0d 100644 --- 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]