fix

mlcommons · Dec 7, 2023 · 5a24308 · 5a24308
1 parent e525726
commit 5a24308
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/models.py b/algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/models.py
@@ -184,7 +184,7 @@ def __init__(self,
         nn.ConvTranspose2d(
             in_chans, out_chans, kernel_size=2, stride=2, bias=False),
         nn.GroupNorm(num_groups=1, num_channels=out_chans, eps=1e-6),
-        norm_layer,
+        norm_layer(out_chans),
         activation_fn,
     )