ani

ANI717 · ANI717 · commit f66d389e184b · 2021-09-19T22:07:19.000-06:00
diff --git a/DCGAN/config.py b/DCGAN/config.py
@@ -9,7 +9,7 @@
 
 
 # Hyperparameters
-DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 IMG_SIZE = (64,64)
 IMG_CHANNELS = 3
diff --git a/WGAN-GP/config.py b/WGAN-GP/config.py
@@ -9,7 +9,7 @@
 
 
 # Hyperparameters
-DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 IMG_SIZE = (64,64)
 IMG_CHANNELS = 3
