Skip to content

Commit d26ab1e

Browse files
virginiafdezVirginia Fernandez
authored andcommitted
Addition of norm_eps parameter to spade_autoencoderkl.py in response to issue Project-MONAI#7946.
Signed-off-by: Virginia Fernandez <[email protected]>
1 parent 378181c commit d26ab1e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

monai/networks/nets/spade_autoencoderkl.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def __init__(
5959
label_nc=label_nc,
6060
norm_nc=in_channels,
6161
norm="GROUP",
62-
norm_params={"num_groups": norm_num_groups, "affine": False},
62+
norm_params={"num_groups": norm_num_groups, "affine": False, "norm_eps": norm_eps},
6363
hidden_channels=spade_intermediate_channels,
6464
kernel_size=3,
6565
spatial_dims=spatial_dims,
@@ -77,7 +77,7 @@ def __init__(
7777
label_nc=label_nc,
7878
norm_nc=out_channels,
7979
norm="GROUP",
80-
norm_params={"num_groups": norm_num_groups, "affine": False},
80+
norm_params={"num_groups": norm_num_groups, "affine": False, "norm_eps": norm_eps},
8181
hidden_channels=spade_intermediate_channels,
8282
kernel_size=3,
8383
spatial_dims=spatial_dims,

0 commit comments

Comments
 (0)