lemonaddie commited on
Commit
a8aa399
·
verified ·
1 Parent(s): 6e2b2c5

Update app_recon.py

Browse files
Files changed (1) hide show
  1. app_recon.py +1 -0
app_recon.py CHANGED
@@ -54,6 +54,7 @@ import torchvision.transforms.functional as TF
54
  from torchvision.transforms import InterpolationMode
55
 
56
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
57
 
58
  # Initialization.. Downloading the ckpts
59
 
 
54
  from torchvision.transforms import InterpolationMode
55
 
56
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
57
+ print("Our current device is ", device)
58
 
59
  # Initialization.. Downloading the ckpts
60