Spaces:

radames
/

PIFu-Clothed-Human-Digitization

Runtime error

radames commited on Apr 27, 2022

Commit

dd12f3e

•

1 Parent(s): b4b55bd

Update PIFu/apps/eval_spaces.py

Files changed (1) hide show

PIFu/apps/eval_spaces.py CHANGED Viewed

@@ -38,7 +38,8 @@ class Evaluator:
         ])
         # set cuda
         cuda = torch.device('cuda:%d' % opt.gpu_id) if torch.cuda.is_available() else torch.device('cpu')
         # create net
         netG = HGPIFuNet(opt, projection_mode).to(device=cuda)
         print('Using Network: ', netG.name)

         ])
         # set cuda
         cuda = torch.device('cuda:%d' % opt.gpu_id) if torch.cuda.is_available() else torch.device('cpu')
+        print("CUDDAAAAA ???",  torch.cuda.get_device_name(0))
         # create net
         netG = HGPIFuNet(opt, projection_mode).to(device=cuda)
         print('Using Network: ', netG.name)