From 54019e4137ecccc9b80883299590a349dcaf6f6e Mon Sep 17 00:00:00 2001 From: Virginia Fernandez <61539159+virginiafdez@users.noreply.github.com> Date: Tue, 30 Jul 2024 08:15:07 +0100 Subject: [PATCH] Addition of norm_eps (#7962) Fixes # . ### Description Addition of norm_eps to spade_autoencoderkl.py as per https://github.com/Project-MONAI/MONAI/issues/7946. ### Types of changes - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [ ] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. --------- Signed-off-by: Virginia Fernandez Signed-off-by: Virginia Fernandez Co-authored-by: Virginia Fernandez Co-authored-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..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,