diff --git a/fastmri/models/feature_varnet.py b/fastmri/models/feature_varnet.py index f9334566..343841d3 100644 --- a/fastmri/models/feature_varnet.py +++ b/fastmri/models/feature_varnet.py @@ -53,9 +53,9 @@ def image_uncrop(image: Tensor, original_image: Tensor) -> Tensor: pad_height_left, pad_width = _calc_uncrop(image.shape[-1], in_shape[-1]) try: - original_image[..., pad_height_top:pad_height, pad_height_left:pad_width] = ( - image[...] - ) + original_image[ + ..., pad_height_top:pad_height, pad_height_left:pad_width + ] = image[...] except RuntimeError: print(f"in_shape: {in_shape}, image shape: {image.shape}") raise @@ -553,7 +553,6 @@ def forward(self, image: torch.Tensor) -> torch.Tensor: class NormUnet(nn.Module): - def __init__( self, chans: int, @@ -644,7 +643,6 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: class Norm1DUnet(nn.Module): - def __init__( self, chans: int,