rohan-paul
diff --git a/‎Aerial-Drone-Images-Semantic-Segmentation-PyTorch-Inception-V4-Pretrained/train.py‎
Lines changed: 1 addition & 1 deletion b/‎Aerial-Drone-Images-Semantic-Segmentation-PyTorch-Inception-V4-Pretrained/train.py‎
Lines changed: 1 addition & 1 deletion
@@ -26,7 +26,7 @@
 from utils import *
 from datagen import *
 
-device = torch.device("cuda" if torch.cuda.is_available else "cpu")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 
 def get_lr(optimizer):