debugging

mlcommons · Dec 6, 2023 · c788d62 · c788d62
1 parent a84d5fa
commit c788d62
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/models.py b/algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/models.py
@@ -115,6 +115,7 @@ def __init__(self,
 
     if use_layer_norm:
       norm_layer = nn.LayerNorm
+      norm_layer = nn.InstanceNorm2d
     else:
       norm_layer = nn.InstanceNorm2d
     if use_tanh: