Spaces:

ericup
/

celldetection

Running on Zero

ericup commited on Mar 11

Commit

d551587

•

1 Parent(s): cb055f2

Gpu stats

Files changed (1) hide show

cpn.py CHANGED Viewed

@@ -58,8 +58,8 @@ class CpnInterface:
         with torch.no_grad():
             out = cd.asnumpy(self.model(x, crop_size=self.tile_size,
                                         stride=max(64, self.tile_size - self.overlap)))
-            if torch.cuda.device_count():
-                print(cd.GpuStats())
         contours, = out['contours']
         boxes, = out['boxes']

         with torch.no_grad():
             out = cd.asnumpy(self.model(x, crop_size=self.tile_size,
                                         stride=max(64, self.tile_size - self.overlap)))
+            # if torch.cuda.device_count():
+            #     print(cd.GpuStats())
         contours, = out['contours']
         boxes, = out['boxes']