We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 56a8a56 commit 4dce9d8Copy full SHA for 4dce9d8
models/brats_mri_generative_diffusion/configs/train_diffusion.json
@@ -17,7 +17,7 @@
17
"load_autoencoder": "$@autoencoder_def.load_old_state_dict(torch.load(@load_autoencoder_path))",
18
"autoencoder": "$@autoencoder_def.to(@device)",
19
"network_def": {
20
- "_target_": "generative.networks.nets.DiffusionModelUNet",
+ "_target_": "monai.networks.nets.diffusion_model_unet.DiffusionModelUNet",
21
"spatial_dims": "@spatial_dims",
22
"in_channels": "@latent_channels",
23
"out_channels": "@latent_channels",
0 commit comments