Skip to content

Commit 7ba43f9

Browse files
committed
Removing instantiation in forward
Signed-off-by: Eric Kerfoot <[email protected]>
1 parent cf5790d commit 7ba43f9

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

monai/networks/nets/diffusion_model_unet.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2063,9 +2063,6 @@ def forward(
20632063
h = h.reshape(h.shape[0], -1)
20642064

20652065
# 5. out
2066-
self.out = nn.Sequential(
2067-
nn.Linear(h.shape[1], 512), nn.ReLU(), nn.Dropout(0.1), nn.Linear(512, self.out_channels)
2068-
)
20692066
output: torch.Tensor = self.out(h)
20702067

20712068
return output

0 commit comments

Comments
 (0)