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

refactor: update common.py

Browse Source
This commit is contained in:
Fam Zheng 2025-12-27 10:33:48 +00:00
parent f5acc55939
commit a40d664e09
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 * 2 results_per_img = ncells * ncells
ret = [] ret = []
with torch.no_grad(): with torch.no_grad():
tensors = [] tensors = []