Skip to content

Commit 4dce9d8

Browse files
committed
Update brats_mri_generative_diffusion to monai1.4
Signed-off-by: Can-Zhao <[email protected]>
1 parent 56a8a56 commit 4dce9d8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/brats_mri_generative_diffusion/configs/train_diffusion.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
"load_autoencoder": "$@autoencoder_def.load_old_state_dict(torch.load(@load_autoencoder_path))",
1818
"autoencoder": "$@autoencoder_def.to(@device)",
1919
"network_def": {
20-
"_target_": "generative.networks.nets.DiffusionModelUNet",
20+
"_target_": "monai.networks.nets.diffusion_model_unet.DiffusionModelUNet",
2121
"spatial_dims": "@spatial_dims",
2222
"in_channels": "@latent_channels",
2323
"out_channels": "@latent_channels",

0 commit comments

Comments
 (0)