Fix num_head_channels (#316)

Signed-off-by: Walter Hugo Lopez Pinaya <[email protected]>
Project-MONAI · Mar 16, 2023 · 9c5c3fd · 9c5c3fd
1 parent 7d02598
commit 9c5c3fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb b/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb
@@ -354,7 +354,7 @@
     "    out_channels=1,\n",
     "    num_channels=[256, 256, 512],\n",
     "    attention_levels=[False, False, True],\n",
-    "    num_head_channels=[256, 256, 512],\n",
+    "    num_head_channels=[0, 0, 512],\n",
     "    num_res_blocks=2,\n",
     ")\n",
     "model.to(device)\n",

diff --git a/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.py b/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.py
@@ -149,7 +149,7 @@
     out_channels=1,
     num_channels=[256, 256, 512],
     attention_levels=[False, False, True],
-    num_head_channels=[256, 256, 512],
+    num_head_channels=[0, 0, 512],
     num_res_blocks=2,
 )
 model.to(device)