Spaces:

ericup
/

celldetection

Running on Zero

ericup commited on Feb 9

Commit

bd5e5cb

•

1 Parent(s): 20394c2

Update cpn.py

Files changed (1) hide show

cpn.py CHANGED Viewed

@@ -36,7 +36,7 @@ class CpnInterface:
         self.device = ('cuda' if torch.cuda.is_available() else 'cpu') if device is None else device
         self.model = cd.models.LitCpn(model).to(device)
         self.model.eval()
-        self.tile_size = 1856
         self.overlap = 384
     def __call__(

         self.device = ('cuda' if torch.cuda.is_available() else 'cpu') if device is None else device
         self.model = cd.models.LitCpn(model).to(device)
         self.model.eval()
+        self.tile_size = 1664
         self.overlap = 384
     def __call__(