diff --git a/options/base_options.py b/options/base_options.py index 35ba46e..9e3c3f1 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -89,7 +89,7 @@ class BaseOptions(): else: print("ERROR") else: - mps_device = torch.device("mps") + torch.device("mps") args = vars(self.opt)