x/themblem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

common.py: restore results_per_img = ncells * ncells * 2

Browse Source
This commit is contained in:
Fam Zheng 2025-12-27 11:04:15 +00:00
parent 8fb224651a
commit e0404ebf5a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
emblem5/ai/common.py
View File

@ -150,7 +150,7 @@ def make_stripe_img(left, right, nstripes):
return ret return ret
def predict_multi(model, transforms, images, ncells=1): def predict_multi(model, transforms, images, ncells=1):
results_per_img = ncells * ncells results_per_img = ncells * ncells * 2
ret = [] ret = []
with torch.no_grad(): with torch.no_grad():
tensors = [] tensors = []