Spaces:
Running
on
Zero
Running
on
Zero
Update cpn.py
Browse files
cpn.py
CHANGED
@@ -54,6 +54,8 @@ class CpnInterface:
|
|
54 |
with torch.no_grad():
|
55 |
out = cd.asnumpy(self.model(x, crop_size=self.tile_size,
|
56 |
stride=max(64, self.tile_size - self.overlap)))
|
|
|
|
|
57 |
|
58 |
contours, = out['contours']
|
59 |
boxes, = out['boxes']
|
|
|
54 |
with torch.no_grad():
|
55 |
out = cd.asnumpy(self.model(x, crop_size=self.tile_size,
|
56 |
stride=max(64, self.tile_size - self.overlap)))
|
57 |
+
if torch.cuda.device_count():
|
58 |
+
print(cd.GpuStats())
|
59 |
|
60 |
contours, = out['contours']
|
61 |
boxes, = out['boxes']
|