From 39acbe038a6d62af0c167e16682ad92b2879549d Mon Sep 17 00:00:00 2001 From: priyakasimbeg Date: Thu, 7 Dec 2023 03:06:19 +0000 Subject: [PATCH] fix --- .../workloads/fastmri/fastmri_pytorch/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/models.py b/algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/models.py index d39b09715..5199e5aa6 100644 --- 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))