Merge pull request #452 from AllenCellModeling/hotfix/fix-plugin-conf…

…ig-saving-callback add predict to saving callback
AllenCellModeling · Nov 25, 2024 · 07959b0 · 07959b0
2 parents ad37fcb + 4247af3
commit 07959b0
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
@@ -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