From 4dce9d8ff6a2a66fae027059dd26c01b5737cbe4 Mon Sep 17 00:00:00 2001 From: Can-Zhao Date: Wed, 21 Aug 2024 22:54:04 +0000 Subject: [PATCH] Update brats_mri_generative_diffusion to monai1.4 Signed-off-by: Can-Zhao --- .../brats_mri_generative_diffusion/configs/train_diffusion.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/brats_mri_generative_diffusion/configs/train_diffusion.json b/models/brats_mri_generative_diffusion/configs/train_diffusion.json index e58b1618..74e702bf 100644 --- 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",