From 36ab45db4c7aa7020506f1d339164bfa4fb06e61 Mon Sep 17 00:00:00 2001 From: Virginia Fernandez Date: Mon, 29 Jul 2024 15:39:40 +0100 Subject: [PATCH 1/2] Addition of norm_eps Signed-off-by: Virginia Fernandez --- monai/networks/nets/spade_autoencoderkl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monai/networks/nets/spade_autoencoderkl.py b/monai/networks/nets/spade_autoencoderkl.py index 294b121c94..c102b2be54 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, "norm_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, "norm_eps": norm_eps}, hidden_channels=spade_intermediate_channels, kernel_size=3, spatial_dims=spatial_dims, From 7a819f28d9eab300cbb0f253d41c3991bda4941b Mon Sep 17 00:00:00 2001 From: Virginia Fernandez Date: Mon, 29 Jul 2024 21:25:17 +0100 Subject: [PATCH 2/2] Addition of norm_eps - modification Signed-off-by: Virginia Fernandez --- monai/networks/nets/spade_autoencoderkl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monai/networks/nets/spade_autoencoderkl.py b/monai/networks/nets/spade_autoencoderkl.py index c102b2be54..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_eps": norm_eps}, + 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_eps": norm_eps}, + norm_params={"num_groups": norm_num_groups, "affine": False, "eps": norm_eps}, hidden_channels=spade_intermediate_channels, kernel_size=3, spatial_dims=spatial_dims,