fix

mlcommons · Dec 5, 2023 · b419962 · b419962
1 parent 71f08ee
commit b419962
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
@@ -61,7 +61,7 @@ def __init__(self,
     self.up_conv.append(
         nn.Sequential(
             ConvBlock(ch * 2, ch, dropout_rate, use_tanh, use_layer_norm),
-            nn.Conv2d(ch, out_chans=1, kernel_size=1, stride=1),
+            nn.Conv2d(ch, 1, kernel_size=1, stride=1),
         ))
 
     for m in self.modules():