ericup commited on
Commit
7da7e9e
1 Parent(s): fb0a5c2

Update cpn.py

Browse files
Files changed (1) hide show
  1. cpn.py +2 -2
cpn.py CHANGED
@@ -36,8 +36,8 @@ class CpnInterface:
36
  self.device = ('cuda' if torch.cuda.is_available() else 'cpu') if device is None else device
37
  self.model = cd.models.LitCpn(model).to(device)
38
  self.model.eval()
39
- self.tile_size = 768
40
- self.overlap = 384
41
 
42
  def __call__(
43
  self,
 
36
  self.device = ('cuda' if torch.cuda.is_available() else 'cpu') if device is None else device
37
  self.model = cd.models.LitCpn(model).to(device)
38
  self.model.eval()
39
+ self.tile_size = 1536
40
+ self.overlap = 1152
41
 
42
  def __call__(
43
  self,