replaces custom saver with SaveImage

Project-MONAI · Jan 21, 2025 · 0fcc2bd · 0fcc2bd
1 parent 1672aa1
commit 0fcc2bd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 32 deletions.
diff --git a/models/brain_image_synthesis_latent_diffusion_model/configs/inference.json b/models/brain_image_synthesis_latent_diffusion_model/configs/inference.json
@@ -105,8 +105,8 @@
     "saver": {
         "_target_": "SaveImage",
         "_requires_": "@create_output_dir",
-        "output_dir": "@output_dir"
+        "output_dir": "@output_dir",
+        "output_postfix": "@out_file"
     },
-    "run": "[email protected](@sample, @out_file)",
-    "save": "$torch.save(@sample, @output_dir + '/' + @out_file + '.pt')"
+    "run": "$@saver(@sample[0][0])"
 }
diff --git a/models/brain_image_synthesis_latent_diffusion_model/scripts/saver.py b/models/brain_image_synthesis_latent_diffusion_model/scripts/saver.py