fix

mlcommons · Dec 7, 2023 · 39acbe0 · 39acbe0
1 parent ed59eb8
commit 39acbe0
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
@@ -52,7 +52,7 @@ def __init__(self,
 
     self.up_conv = nn.ModuleList()
     self.up_transpose_conv = nn.ModuleList()
-    size = size * 2
+    # size = size * 2
     for _ in range(num_pool_layers - 1):
       self.up_transpose_conv.append(
           TransposeConvBlock(ch * 2, ch, use_tanh, use_layer_norm, size))