Update fs_model.py

2022-07-13 18:42:20 +02:00
parent c01d582b3f
commit 2389ebe35d
1 changed files with 1 additions and 1 deletions
--- a/models/fs_model.py
+++ b/models/fs_model.py
@@ -48,7 +48,7 @@ class fsModel(BaseModel):
            torch.backends.cudnn.benchmark = True
        self.isTrain = opt.isTrain

-          if not torch.backends.mps.is_available():
+        if not torch.backends.mps.is_available():
           if not torch.backends.mps.is_built():
             device = torch.device("cuda:0")
           else: