Re-adding fastmri/models/feature_varnet.py to ensure CI passes

facebookresearch · Jun 12, 2024 · dba55c6 · dba55c6
1 parent c0ba931
commit dba55c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastmri/models/feature_varnet.py b/fastmri/models/feature_varnet.py
@@ -938,7 +938,7 @@ def forward(
             kspace_pred = cascade(
                 kspace_pred, feature_image.ref_kspace, mask, feature_image.sens_maps
             )
-        # Return Final Image
+        # Divide with k-space factor and Return Final Image
         kspace_pred = kspace_pred / self.kspace_mult_factor
         return rss(complex_abs(ifft2c(kspace_pred)), dim=1)