refactor: update common.py

2025-12-27 10:33:48 +00:00 · 2025-12-27 10:33:48 +00:00 · a40d664e09
commit a40d664e09
parent f5acc55939
1 changed files with 1 additions and 1 deletions
--- a/emblem5/ai/common.py
+++ b/emblem5/ai/common.py
@ -150,7 +150,7 @@ def make_stripe_img(left, right, nstripes):
    return ret

 def predict_multi(model, transforms, images, ncells=1):
-    results_per_img = ncells * ncells * 2
+    results_per_img = ncells * ncells
    ret = []
    with torch.no_grad():
        tensors = []