From 3a0456337a1ff6d8bb8bdecaa244275b024b99b1 Mon Sep 17 00:00:00 2001 From: Walter Hugo Lopez Pinaya Date: Thu, 16 Mar 2023 22:41:15 +0000 Subject: [PATCH] Fix num_head_channels Signed-off-by: Walter Hugo Lopez Pinaya --- tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb | 2 +- tutorials/generative/3d_ddpm/3d_ddpm_tutorial.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb b/tutorials/generative/3d_ddpm/3d_ddpm_tutorial.ipynb index 0c3ff8a5..188d78d5 100644 --- 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 index 612b84dd..41c4960f 100644 --- 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)