From 4247af3f8e70d0a6ce627c6736cc26b8a31993bc Mon Sep 17 00:00:00 2001 From: "brian.kim" Date: Mon, 25 Nov 2024 10:28:01 -0800 Subject: [PATCH] add predict to saving callback --- 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 deb8513d..e89c9b58 100644 --- a/configs/experiment/im2im/segmentation_plugin.yaml +++ b/configs/experiment/im2im/segmentation_plugin.yaml @@ -78,5 +78,5 @@ callbacks: _target_: cyto_dl.callbacks.ImageSaver save_dir: ${paths.output_dir} save_every_n_epochs: ${model.save_images_every_n_epochs} - stages: ["train", "test", "val"] + stages: ["train", "test", "val", "predict"] save_input: True