Move module imports

facebookresearch · Jul 23, 2024 · c5eaf17 · c5eaf17
1 parent 46a4c0b
commit c5eaf17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fastmri_examples/feature_varnet/feature_varnet_module.py b/fastmri_examples/feature_varnet/feature_varnet_module.py
@@ -9,14 +9,14 @@
 from argparse import ArgumentParser
 
 import torch
-
-torch.set_float32_matmul_precision("high")
 from feature_varnet import FIVarNet
 
 from fastmri.data.transforms import center_crop, center_crop_to_smallest
 from fastmri.losses import SSIMLoss
 from fastmri.pl_modules.mri_module import MriModule
 
+torch.set_float32_matmul_precision("high")
+
 
 class FIVarNetModule(MriModule):
     def __init__(