Update brats_mri_generative_diffusion to monai1.4

Signed-off-by: Can-Zhao <[email protected]>
Project-MONAI · Aug 21, 2024 · 4dce9d8 · 4dce9d8
1 parent 56a8a56
commit 4dce9d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/brats_mri_generative_diffusion/configs/train_diffusion.json b/models/brats_mri_generative_diffusion/configs/train_diffusion.json
@@ -17,7 +17,7 @@
     "load_autoencoder": "$@autoencoder_def.load_old_state_dict(torch.load(@load_autoencoder_path))",
     "autoencoder": "$@autoencoder_def.to(@device)",
     "network_def": {
-        "_target_": "generative.networks.nets.DiffusionModelUNet",
+        "_target_": "monai.networks.nets.diffusion_model_unet.DiffusionModelUNet",
         "spatial_dims": "@spatial_dims",
         "in_channels": "@latent_channels",
         "out_channels": "@latent_channels",