Spaces:

shivambhosale
/

spacenet3-unet-1024-1024

Runtime error

shivambhosale commited on Jul 6, 2022

Commit

3904368

•

1 Parent(s): ad4887b

Update UNet.py

Files changed (1) hide show

UNet.py CHANGED Viewed

@@ -1,8 +1,4 @@
-import torch # ; print('Using torch version -', torch.__version__)
-if torch.cuda.is_available():
-    device = 'cuda'
-else:
-    device = 'cpu'
 from torch.nn import Module, Conv2d, ReLU, ModuleList, MaxPool2d, ConvTranspose2d, BCELoss, BCEWithLogitsLoss, functional as F
 from torch.optim import Adam
 from torchvision import transforms

+import torch
 from torch.nn import Module, Conv2d, ReLU, ModuleList, MaxPool2d, ConvTranspose2d, BCELoss, BCEWithLogitsLoss, functional as F
 from torch.optim import Adam
 from torchvision import transforms