diff --git a/monai/networks/nets/spade_autoencoderkl.py b/monai/networks/nets/spade_autoencoderkl.py index 294b121c94..d5794a9227 100644 --- a/monai/networks/nets/spade_autoencoderkl.py +++ b/monai/networks/nets/spade_autoencoderkl.py @@ -59,7 +59,7 @@ def __init__( label_nc=label_nc, norm_nc=in_channels, norm="GROUP", - norm_params={"num_groups": norm_num_groups, "affine": False}, + norm_params={"num_groups": norm_num_groups, "affine": False, "eps": norm_eps}, hidden_channels=spade_intermediate_channels, kernel_size=3, spatial_dims=spatial_dims, @@ -77,7 +77,7 @@ def __init__( label_nc=label_nc, norm_nc=out_channels, norm="GROUP", - norm_params={"num_groups": norm_num_groups, "affine": False}, + norm_params={"num_groups": norm_num_groups, "affine": False, "eps": norm_eps}, hidden_channels=spade_intermediate_channels, kernel_size=3, spatial_dims=spatial_dims,