From 2389ebe35d230d2ecfedafe77f32b524aca6289a Mon Sep 17 00:00:00 2001 From: Smiril Date: Wed, 13 Jul 2022 18:42:20 +0200 Subject: [PATCH] Update fs_model.py --- models/fs_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/fs_model.py b/models/fs_model.py index c54a28d..bd4a671 100644 --- 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: