fix

mlcommons · Dec 7, 2023 · 7ac38b1 · 7ac38b1
1 parent 268521a
commit 7ac38b1
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
@@ -62,7 +62,7 @@ def __init__(self,
       ch //= 2
 
     self.up_transpose_conv.append(
-        TransposeConvBlock(ch * 2, ch, use_tanh, use_layer_norm))
+        TransposeConvBlock(ch * 2, ch, use_tanh, use_layer_norm, size))
     size = int(size * 2)
     self.up_conv.append(
         nn.Sequential(