Closed
Description
Describe the bug
In the init
of SPADEResBlock
norm_eps
was defined but not used in the class methods.
To Reproduce
Steps to reproduce the behavior:
None
Expected behavior
The SPADE layer should take the eps
for the GroupNorm:
Current:
self.norm1 = SPADE(
label_nc=label_nc,
norm_nc=in_channels,
norm="GROUP",
norm_params={"num_groups": norm_num_groups, "affine": False},
hidden_channels=spade_intermediate_channels,
kernel_size=3,
spatial_dims=spatial_dims,
)
Expected:
self.norm1 = SPADE(
label_nc=label_nc,
norm_nc=in_channels,
norm="GROUP",
norm_params={"num_groups": norm_num_groups, "affine": False, "eps": norm_eps},
hidden_channels=spade_intermediate_channels,
kernel_size=3,
spatial_dims=spatial_dims,
)