Update cpn.py

cpn.py

@@ -54,6 +54,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']